Krishna086 commited on
Commit
700ac54
·
verified ·
1 Parent(s): e2b9d0c

Update translation.py

Browse files
Files changed (1) hide show
  1. 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):