Spaces:
Runtime error
Runtime error
youssef1214
commited on
Commit
•
56cd27c
1
Parent(s):
7b86f88
Update main.py
Browse files
main.py
CHANGED
@@ -31,9 +31,9 @@ import firebase_admin
|
|
31 |
from firebase_admin import credentials
|
32 |
from firebase_admin import firestore
|
33 |
from transformers import BertTokenizer, AutoModelForSeq2SeqLM, pipeline
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
# Firebase ininlaziton
|
38 |
cred = credentials.Certificate(
|
39 |
"text-to-emotions-firebase-adminsdk-8isbn-dffbdf01e8.json")
|
@@ -41,11 +41,11 @@ firebase_admin.initialize_app(cred)
|
|
41 |
|
42 |
# Model summury
|
43 |
model_name="abdalrahmanshahrour/auto-arabic-summarization"
|
44 |
-
|
45 |
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
|
50 |
# Model inilization
|
51 |
isristemmer = ISRIStemmer()
|
|
|
31 |
from firebase_admin import credentials
|
32 |
from firebase_admin import firestore
|
33 |
from transformers import BertTokenizer, AutoModelForSeq2SeqLM, pipeline
|
34 |
+
from arabert.preprocess import ArabertPreprocessor
|
35 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
36 |
+
import re
|
37 |
# Firebase ininlaziton
|
38 |
cred = credentials.Certificate(
|
39 |
"text-to-emotions-firebase-adminsdk-8isbn-dffbdf01e8.json")
|
|
|
41 |
|
42 |
# Model summury
|
43 |
model_name="abdalrahmanshahrour/auto-arabic-summarization"
|
44 |
+
preprocessor = ArabertPreprocessor(model_name="")
|
45 |
|
46 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
47 |
+
modelsummary =from_pretrained_keras(model_name)
|
48 |
+
pipeline1 = pipeline("text2text-generation",model=modelsummary,tokenizer=tokenizer)
|
49 |
|
50 |
# Model inilization
|
51 |
isristemmer = ISRIStemmer()
|