youssef1214 commited on
Commit
074ac5f
1 Parent(s): 31b644c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -7
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
- 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,11 +41,11 @@ firebase_admin.initialize_app(cred)
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 = AutoModelForSeq2SeqLM.from_pretrained(model_name)
48
- pipeline1 = pipeline("text2text-generation",model=modelsummary,tokenizer=tokenizer)
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()