Update src/Translate.py

src/Translate.py

@@ -42,11 +42,14 @@ 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)
-        # 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)
+        tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+        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)
         # pipe = pipeline("translation", model="facebook/mbart-large-cc25")
+        # translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=self.sl, tgt_lang=self.tl)
+        # translated_text = translator(text, max_length=512)
+        # return translated_text[0]['translation_text']
         # Tokenize and translate
         inputs = tokenizer(self.input_text, return_tensors="pt")
         translated_tokens = model.generate(