Update src/Translate.py

src/Translate.py

@@ -42,7 +42,7 @@ class Translators:
         tgt_lang = f"{self.tl}_{self.tl.upper()}"
         # Load model and tokenizer
         # from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-        tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+        tokenizer = AutoTokenizer.from_pretrained(self.model_name, src_lang=src_lang)
         model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
         # model = MBartForConditionalGeneration.from_pretrained(self.model_name)
         # tokenizer = MBartTokenizer.from_pretrained(self.model_name, src_lang=src_lang)