TiberiuCristianLeon commited on
Commit
6dc437d
·
verified ·
1 Parent(s): cae0132

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")