Update inference.py

medahmedkrichen · Dec 15, 2024 · cf1c475 · cf1c475
1 parent b544b90
commit cf1c475
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/inference.py b/inference.py
@@ -374,21 +374,17 @@ def __init__(self, Video_path, source_language, target_language,
                         'None': None}
 
 
-
         if self.source_language == 'tr':
             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)