Update app.py
Browse files
app.py
CHANGED
@@ -289,15 +289,14 @@ model.eval()
|
|
289 |
bot_name = "WeASK"
|
290 |
|
291 |
###removed
|
292 |
-
from transformers import MBartForConditionalGeneration,
|
293 |
|
294 |
-
def download_model():
|
295 |
-
|
296 |
-
|
297 |
-
|
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 |
|