Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -24,6 +24,11 @@ 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 == "ArissBandoss/mos2fr-3B":
+        peft_config = PeftConfig.from_pretrained("ArissBarndoss/mos2fr-3B")
+        base_model = AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path)
+        model = PeftModel.from_pretrained(base_model, "ArissBarndoss/mos2fr-3B")
         
     trans_pipe = pipeline("translation", 
                           model=model, tokenizer=tokenizer,