Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -39,16 +39,11 @@ def goai_traduction(text, src_lang, tgt_lang):
     outputs = model.generate(
         **inputs,
         forced_bos_token_id=tgt_lang_id,
+        eos_token_id=tokenizer.eos_token_id
         max_length=512,
-        length_penalty=1.0,
            num_beams=4,
-    do_sample=True,
-    temperature=1.0,
-    top_k=50,
-    top_p=0.95,
-    repetition_penalty=1.2,
-    no_repeat_ngram_size=3,
-      eos_token_id=None
+    do_sample=False,
+    no_repeat_ngram_size=3
     )
 
     print("Token IDs:", outputs)