Update translation.py

translation.py

@@ -20,16 +20,20 @@ def load_model(source_lang, target_lang):
     try:
         if source_lang == target_lang:
             return _load_default_model()
+        # Try direct model first
         model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
-        tokenizer = MarianTokenizer.from_pretrained(model_name)
-        model = MarianMTModel.from_pretrained(model_name)
-        return tokenizer, model
+        try:
+            tokenizer = MarianTokenizer.from_pretrained(model_name)
+            model = MarianMTModel.from_pretrained(model_name)
+            return tokenizer, model
+        except Exception:
+            # Pivot through English for non-English pairs
+            if source_lang != "en" and target_lang != "en":
+                en_to_target = load_model("en", target_lang)
+                source_to_en = load_model(source_lang, "en")
+                return source_to_en if source_lang == "en" else en_to_target
+            return _load_default_model()
     except Exception:
-        # Pivot through English if direct model fails
-        if source_lang != "en":
-            return load_model("en", target_lang)
-        elif target_lang != "en":
-            return load_model(source_lang, "en")
         return _load_default_model()
 
 def translate(text, source_lang, target_lang):