Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ class Translators:
|
|
| 102 |
translated_text = error
|
| 103 |
return translated_text
|
| 104 |
|
| 105 |
-
def
|
| 106 |
try:
|
| 107 |
pipe = pipeline("translation", model=self.model_name, device=self.device)
|
| 108 |
iso_dict = {iso[1]: iso[3] for iso in non_empty_isos}
|
|
@@ -400,8 +400,8 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
|
|
| 400 |
message_text = f'Translated from {s_language} to {t_language} with {model_name}'
|
| 401 |
translated_text = None
|
| 402 |
try:
|
| 403 |
-
if model_name
|
| 404 |
-
translated_text, message_text = Translators(model_name, sl, tl, input_text).
|
| 405 |
|
| 406 |
elif model_name == "Helsinki-NLP":
|
| 407 |
translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
|
|
|
|
| 102 |
translated_text = error
|
| 103 |
return translated_text
|
| 104 |
|
| 105 |
+
def HelsinkiNLP_mul(self):
|
| 106 |
try:
|
| 107 |
pipe = pipeline("translation", model=self.model_name, device=self.device)
|
| 108 |
iso_dict = {iso[1]: iso[3] for iso in non_empty_isos}
|
|
|
|
| 400 |
message_text = f'Translated from {s_language} to {t_language} with {model_name}'
|
| 401 |
translated_text = None
|
| 402 |
try:
|
| 403 |
+
if "-mul" in model_name.lower() or "mul-" in model_name.lower():
|
| 404 |
+
translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mul()
|
| 405 |
|
| 406 |
elif model_name == "Helsinki-NLP":
|
| 407 |
translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
|