youssef1214 commited on
Commit
56cd27c
1 Parent(s): 7b86f88

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 =from_pretrained_keras(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()