Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -188,11 +188,12 @@ def translate_text(input_text: str, sselected_language: str, tselected_language:
|
|
| 188 |
model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name))
|
| 189 |
except EnvironmentError as error:
|
| 190 |
return f"Error finding model: {model_name}! Try other available language combination.", error
|
| 191 |
-
|
|
|
|
| 192 |
translated_text = eurollm_instruct(model_name, sselected_language, tselected_language, input_text)
|
| 193 |
return translated_text, message_text
|
| 194 |
|
| 195 |
-
if model_name
|
| 196 |
translated_text = eurollm(model_name, sselected_language, tselected_language, input_text)
|
| 197 |
return translated_text, message_text
|
| 198 |
|
|
|
|
| 188 |
model = model_to_cuda(AutoModelForSeq2SeqLM.from_pretrained(model_name))
|
| 189 |
except EnvironmentError as error:
|
| 190 |
return f"Error finding model: {model_name}! Try other available language combination.", error
|
| 191 |
+
|
| 192 |
+
if model_name == "utter-project/EuroLLM-1.7B-Instruct":
|
| 193 |
translated_text = eurollm_instruct(model_name, sselected_language, tselected_language, input_text)
|
| 194 |
return translated_text, message_text
|
| 195 |
|
| 196 |
+
if model_name == "utter-project/EuroLLM-1.7B":
|
| 197 |
translated_text = eurollm(model_name, sselected_language, tselected_language, input_text)
|
| 198 |
return translated_text, message_text
|
| 199 |
|