Update src/Translate.py

src/Translate.py

@@ -71,9 +71,10 @@ class Translators:
         
         # tell tokenizer the source language
         tokenizer.src_lang = "en_XX"
+        tokenizer.tgt_lang = "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"]
+        # 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)
@@ -83,8 +84,7 @@ class Translators:
         result = pipe(
             src_text,
             num_beams=4,
-            max_length=512,
-            forced_bos_token_id=forced_bos_token_id
+            max_length=512
         )
         
         return result[0]["translation_text"], self.message