Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -22,7 +22,7 @@ def goai_traduction(text, src_lang, tgt_lang):
     else:
         model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
     
-    tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
+    tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token, truncation=True, max_length=512)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
     
     # Ajout du code de langue source