Update src/Translate.py

src/Translate.py

@@ -79,11 +79,13 @@ 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_name)
+        pipe = pipeline("translation", model=model_name, src_lang="en_XX", tgt_lang="ro_RO")
         # "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
-        src_text = "Check general exterior condition."
+        # src_lang (str, optional) — The language of the input.
+        # tgt_lang (str, optional) — The language of the desired output. Might be required for multilingual models. Will not have any effect for single pair translation models
+        src_text = "Check general exterior condition"
         result = pipe(
             src_text,
             num_beams=4,