satyaiyer commited on
Commit
44a9f48
·
verified ·
1 Parent(s): 7f80a0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,9 +9,10 @@ pipe = pipeline("text2text-generation", model="facebook/mbart-large-50-many-to-o
9
  # Load model directly
10
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
11
 
12
- tokenizer = AutoTokenizer.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
13
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
14
 
 
15
  article_hi = "संयुक्त राष्ट्र के प्रमुख का कहना है कि सीरिया में कोई सैन्य समाधान नहीं है"
16
  article_ar = "الأمين العام للأمم المتحدة يقول إنه لا يوجد حل عسكري في سوريا."
17
 
 
9
  # Load model directly
10
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
11
 
12
+ tokenizer = AutoTokenizer.from_pretrained("facebook/mbart-large-50-many-to-one-mmt", use_fast=False)
13
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
14
 
15
+
16
  article_hi = "संयुक्त राष्ट्र के प्रमुख का कहना है कि सीरिया में कोई सैन्य समाधान नहीं है"
17
  article_ar = "الأمين العام للأمم المتحدة يقول إنه لا يوجد حل عسكري في سوريا."
18