Update inference.py

medahmedkrichen · web-flow · commit d26a461948ff · 2024-12-15T01:32:35.000+01:00
diff --git a/inference.py b/inference.py
@@ -375,8 +375,13 @@ def __init__(self, Video_path, source_language, target_language,
     
 
 
-         # 
-        model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-{self.target_language}"
+        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}"
+		
         tokenizer = MarianTokenizer.from_pretrained(model_name)
         model = MarianMTModel.from_pretrained(model_name).to(device)
 
