ipvikas commited on
Commit
5662411
1 Parent(s): 3158dd4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -289,15 +289,14 @@ model.eval()
289
  bot_name = "WeASK"
290
 
291
  ###removed
292
- from transformers import MBartForConditionalGeneration, MBart50Tokenizer
293
 
294
- def download_model():
295
- model_name = "facebook/mbart-large-50-many-to-many-mmt"
296
- model = MBartForConditionalGeneration.from_pretrained(model_name)
297
- tokenizer = MBart50Tokenizer.from_pretrained(model_name)
298
- return model, tokenizer
299
 
300
- model, tokenizer = download_model()
301
 
302
 
303
 
 
289
  bot_name = "WeASK"
290
 
291
  ###removed
292
+ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
293
 
294
+ #def download_model():
295
+ model_name = "facebook/mbart-large-50-many-to-many-mmt"
296
+ model = MBartForConditionalGeneration.from_pretrained(model_name)
297
+ tokenizer = MBart50Tokenizer.from_pretrained(model_name)
 
298
 
299
+ #model, tokenizer = download_model()
300
 
301
 
302