Update src/Translate.py

src/Translate.py

@@ -66,8 +66,8 @@ class Translators:
         model_name = "facebook/mbart-large-cc25"
         
         # load tokenizer and model
-        tokenizer = AutoTokenizer.from_pretrained(model_name)
-        model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+        # tokenizer = AutoTokenizer.from_pretrained(model_name)
+        # model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
         
         # tell tokenizer the source language
         # tokenizer.src_lang = "en_XX"
@@ -79,7 +79,7 @@ class Translators:
         # forced_bos_token_id = tokenizer.lang_code_to_id["ro_RO"]
         
         # create the pipeline (pass tokenizer and model explicitly)
-        pipe = pipeline("translation_en_to_ro", model=model, tokenizer=tokenizer)
+        pipe = pipeline("translation_en_to_ro", model=model_name)
         # "translation" task was used, instead of "translation_XX_to_YY", defaulting to "translation_en_to_ro"
         
         # call the pipeline; generation kwargs are forwarded to model.generate