Update app.py

medahmedkrichen · web-flow · commit b544b90c6f23 · 2024-12-15T02:29:34.000+01:00
diff --git a/app.py b/app.py
@@ -361,15 +361,13 @@ def __init__(self, Video_path, source_language, target_language,
             model_name = f"Helsinki-NLP/opus-mt-trk-{self.target_language}"
         elif self.target_language == 'tr':
             model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-trk"
-        else:
-            model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-{self.target_language}"
-
-        if self.source_language == 'zh-cn':
+        elif self.source_language == 'zh-cn':
             model_name = f"Helsinki-NLP/opus-mt-zh-{self.target_language}"
         elif self.target_language == 'zh-cn':
             model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-zh"
         else:
             model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-{self.target_language}"
+
 		
         tokenizer = MarianTokenizer.from_pretrained(model_name)
         model = MarianMTModel.from_pretrained(model_name).to(device)
