youssef1214 commited on
Commit
163695f
1 Parent(s): 56cd27c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -44,7 +44,7 @@ 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
 
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