TiberiuCristianLeon commited on
Commit
aab3949
·
verified ·
1 Parent(s): a2857f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
25
  return f"Error finding model: {model_name_full}! Try other available language combination.", error
26
  elif model_name.startswith('facebook/nllb'):
27
  from languagecodes import nllb_language_codes
28
- tokenizer = AutoTokenizer.from_pretrained(model_name, token=True, src_lang=nllb_language_codes[sselected_language])
29
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name, token=True)
30
  translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=nllb_language_codes[sselected_language], tgt_lang=nllb_language_codes[tselected_language])
31
  translated_text = translator(input_text, max_length=360)
32
  return translated_text[0]['translation_text'], message_text
 
25
  return f"Error finding model: {model_name_full}! Try other available language combination.", error
26
  elif model_name.startswith('facebook/nllb'):
27
  from languagecodes import nllb_language_codes
28
+ tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang=nllb_language_codes[sselected_language])
29
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
30
  translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=nllb_language_codes[sselected_language], tgt_lang=nllb_language_codes[tselected_language])
31
  translated_text = translator(input_text, max_length=360)
32
  return translated_text[0]['translation_text'], message_text