Update src/Translate.py

src/Translate.py

@@ -70,16 +70,16 @@ class Translators:
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
         
         # tell tokenizer the source language
-        tokenizer.src_lang = "en_XX"
+        # tokenizer.src_lang = "en_XX"
         # tokenizer.tgt_lang = "ro_RO"
         # set the target language as the model's forced BOS token so pipeline will use it implicitly
-        model.config.forced_bos_token_id = tokenizer.lang_code_to_id["ro_RO"]
+        # model.config.forced_bos_token_id = tokenizer.lang_code_to_id["ro_RO"]
         
         # find the id for the target language and force it at generation
         # forced_bos_token_id = tokenizer.lang_code_to_id["ro_RO"]
         
         # create the pipeline (pass tokenizer and model explicitly)
-        pipe = pipeline("translation", model=model, tokenizer=tokenizer)
+        pipe = pipeline("translation_en_to_ro", model=model, tokenizer=tokenizer)
         # "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