Maram-almasary commited on
Commit
3fb74d1
·
verified ·
1 Parent(s): bf158c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,10 +4,13 @@ import graphviz
4
  from PIL import Image
5
  import sentencepiece
6
 
 
 
7
  model_name = "csebuetnlp/mT5_multilingual_XLSum"
8
- tokenizer = AutoTokenizer.from_pretrained(model_name)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
10
 
 
11
  question_generator = pipeline("text2text-generation", model="valhalla/t5-small-e2e-qg")
12
 
13
  def summarize_text(text, src_lang):
 
4
  from PIL import Image
5
  import sentencepiece
6
 
7
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
8
+
9
  model_name = "csebuetnlp/mT5_multilingual_XLSum"
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False) # المحول البطيء
11
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
12
 
13
+
14
  question_generator = pipeline("text2text-generation", model="valhalla/t5-small-e2e-qg")
15
 
16
  def summarize_text(text, src_lang):