TiberiuCristianLeon commited on
Commit
ad4a802
·
verified ·
1 Parent(s): 9b5378d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -44,6 +44,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
44
  sl = all_langs[sselected_language]
45
  tl = all_langs[tselected_language]
46
  message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
 
47
  if model_name == "Helsinki-NLP":
48
  try:
49
  model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
@@ -56,9 +57,9 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
56
  model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name))
57
  except EnvironmentError as error:
58
  return f"Error finding model: {model_name}! Try other available language combination.", error
59
- if 'eurollm' in model_name:
60
  translated_text = eurollm(model_name, sl, tl, input_text)
61
- return translated_text[0]['translation_text'], message_text
62
 
63
  if 'nllb' in model_name:
64
  tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang=languagecodes.nllb_language_codes[sselected_language])
 
44
  sl = all_langs[sselected_language]
45
  tl = all_langs[tselected_language]
46
  message_text = f'Translated from {sselected_language} to {tselected_language} with {model_name}'
47
+ print(message_text)
48
  if model_name == "Helsinki-NLP":
49
  try:
50
  model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
 
57
  model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name))
58
  except EnvironmentError as error:
59
  return f"Error finding model: {model_name}! Try other available language combination.", error
60
+ if 'eurollm' in model_name.lower():
61
  translated_text = eurollm(model_name, sl, tl, input_text)
62
+ return translated_text, message_text
63
 
64
  if 'nllb' in model_name:
65
  tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang=languagecodes.nllb_language_codes[sselected_language])