Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
|
|
44 |
translated_text = translator(input_text, max_length=512)
|
45 |
return translated_text[0]['translation_text'], message_text
|
46 |
|
47 |
-
if model_name.startswith('facebook/mbart-large
|
48 |
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
49 |
model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
50 |
tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
|
|
44 |
translated_text = translator(input_text, max_length=512)
|
45 |
return translated_text[0]['translation_text'], message_text
|
46 |
|
47 |
+
if model_name.startswith('facebook/mbart-large'):
|
48 |
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
49 |
model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
50 |
tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|