Update app.py
Browse files
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
|
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])
|