Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -26,10 +26,10 @@ def goai_traduction(text, src_lang, tgt_lang):
     tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
     model     = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token)
 
-    if model_id == "ArissBarndoss/mos2fr-3B":
-        peft_config = PeftConfig.from_pretrained("ArissBandoss/mos2fr-3B")
+    if model_id == "ArissBandoss/mos2fr-5B-800":
+        peft_config = PeftConfig.from_pretrained("ArissBandoss/mos2fr-5B-800")
         base_model = AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path)
-        model = PeftModel.from_pretrained(base_model, "ArissBandoss/mos2fr-3B")
+        model = PeftModel.from_pretrained(base_model, "ArissBandoss/mos2fr-5B-800")
         
         # Instead of using the pipeline, do direct generation
         tokenizer = AutoTokenizer.from_pretrained(peft_config.base_model_name_or_path)