Update translation.py
Browse files- translation.py +5 -0
translation.py
CHANGED
@@ -25,6 +25,11 @@ def load_model(source_lang, target_lang):
|
|
25 |
model = MarianMTModel.from_pretrained(model_name)
|
26 |
return tokenizer, model
|
27 |
except Exception:
|
|
|
|
|
|
|
|
|
|
|
28 |
return _load_default_model()
|
29 |
|
30 |
def translate(text, source_lang, target_lang):
|
|
|
25 |
model = MarianMTModel.from_pretrained(model_name)
|
26 |
return tokenizer, model
|
27 |
except Exception:
|
28 |
+
# Pivot through English if direct model fails
|
29 |
+
if source_lang != "en":
|
30 |
+
return load_model("en", target_lang)
|
31 |
+
elif target_lang != "en":
|
32 |
+
return load_model(source_lang, "en")
|
33 |
return _load_default_model()
|
34 |
|
35 |
def translate(text, source_lang, target_lang):
|