Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -18,7 +18,7 @@ def goai_traduction(text, src_lang, tgt_lang):
         model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
         
     elif src_lang == "mos_Latn" and tgt_lang == "fra_Latn":
-        model_id = "ArissBandoss/mos2fr-3B"
+        model_id = "ArissBandoss/mos2fr-5B-800"
 
     else:
         model_id = "ArissBandoss/nllb-200-distilled-600M-finetuned-fr-to-mos-V4"
@@ -32,6 +32,7 @@ def goai_traduction(text, src_lang, tgt_lang):
                           model=model, tokenizer=tokenizer, 
                           src_lang=src_lang, tgt_lang=tgt_lang, 
                           max_length=max_length,
+                           max_new_tokens=512,
                           device=device
                          )