Update src/Translate.py

src/Translate.py

@@ -79,7 +79,9 @@ class Translators:
         # forced_bos_token_id = tokenizer.lang_code_to_id["ro_RO"]
         
         # create the pipeline (pass tokenizer and model explicitly)
-        pipe = pipeline("translation", model=model_name, src_lang="en_XX", tgt_lang="ro_RO")
+        # export langs=ar_AR,cs_CZ,de_DE,en_XX,es_XX,et_EE,fi_FI,fr_XX,gu_IN,hi_IN,it_IT,ja_XX,kk_KZ,ko_KR,lt_LT,lv_LV,my_MM,ne_NP,nl_XX,ro_RO,ru_RU,si_LK,tr_TR,vi_VN,zh_CN
+        # pipe = pipeline("translation", model=model_name, src_lang="en_XX", tgt_lang="ro_RO")
+        pipe = pipeline("translation_en_to_de", model="facebook/mbart-large-cc25")
         # "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