TiberiuCristianLeon commited on
Commit
988d5ac
·
verified ·
1 Parent(s): fc650b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
24
  except EnvironmentError as error:
25
  return f"Error finding model: {model_name_full}! Try other available language combination.", error
26
  elif model_name.startswith('facebook/nllb'):
27
- from languagecode import nllb_language_codes
28
  tokenizer = AutoTokenizer.from_pretrained(model_name)
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])
 
24
  except EnvironmentError as error:
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)
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])