Update src/Translate.py

src/Translate.py

@@ -37,12 +37,16 @@ class Translators:
         translation  = pipeline('translation', model = self.model_name)
         return translation(self.input_text)[0]['translation_text'], self.message
     def mbartlarge(self):
-        from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
+        from transformers import MBartForConditionalGeneration, MBart50TokenizerFast, MBartTokenizer
         src_lang = f"{self.sl}_XX"
         tgt_lang = f"{self.tl}_{self.tl.upper()}"
         # Load model and tokenizer
+        # from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+        # tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+        # model = AutoModelForSeq2SeqLM.from_pretrained(self.model_name)
         model = MBartForConditionalGeneration.from_pretrained(self.model_name)
-        tokenizer = MBart50TokenizerFast.from_pretrained(self.model_name, src_lang=src_lang)
+        tokenizer = MBartTokenizer.from_pretrained(self.model_name, src_lang=src_lang)
+        # pipe = pipeline("translation", model="facebook/mbart-large-cc25")
         # Tokenize and translate
         inputs = tokenizer(self.input_text, return_tensors="pt")
         translated_tokens = model.generate(