Spaces:
Running
Running
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])
|