Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -38,12 +38,17 @@ def goai_traduction(text, src_lang, tgt_lang):
     # Génération avec paramètres améliorés
     outputs = model.generate(
         **inputs,
-        forced_bos_token_id=tgt_lang_id,  
-        max_new_tokens=1024,
-        min_length=10,
+        forced_bos_token_id=tgt_lang_id,
+        max_length=512,
         length_penalty=1.0,
-         early_stopping=False,
-        no_repeat_ngram_size=0  # Désactive la pénalité pour les répétitions
+           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
     )
 
     print("Token IDs:", outputs)