Update main.py

main.py

@@ -31,9 +31,9 @@ import firebase_admin
 from firebase_admin import credentials
 from firebase_admin import firestore
 from transformers import BertTokenizer, AutoModelForSeq2SeqLM, pipeline
-from arabert.preprocess import ArabertPreprocessor
-from transformers import AutoTokenizer, AutoModelForCausalLM
-import re
+# from arabert.preprocess import ArabertPreprocessor
+# from transformers import AutoTokenizer, AutoModelForCausalLM
+# import re
 # Firebase ininlaziton
 cred = credentials.Certificate(
     "text-to-emotions-firebase-adminsdk-8isbn-dffbdf01e8.json")
@@ -41,11 +41,11 @@ firebase_admin.initialize_app(cred)
 
 # Model summury
 model_name="abdalrahmanshahrour/auto-arabic-summarization"
-preprocessor = ArabertPreprocessor(model_name="")
+# preprocessor = ArabertPreprocessor(model_name="")
 
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-modelsummary = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-pipeline1 = pipeline("text2text-generation",model=modelsummary,tokenizer=tokenizer)
+# tokenizer = AutoTokenizer.from_pretrained(model_name)
+modelsummary =from_pretrained_keras(model_name)
+# pipeline1 = pipeline("text2text-generation",model=modelsummary,tokenizer=tokenizer)
 
 # Model inilization
 isristemmer = ISRIStemmer()