Update goai_helpers/goai_traduction.py

goai_helpers/goai_traduction.py

@@ -27,9 +27,22 @@ def goai_traduction(text, src_lang, tgt_lang):
     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")
+        try:
+            peft_config = PeftConfig.from_pretrained("ArissBandoss/mos2fr-3B")
+            base_model = AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path)
+            model = PeftModel.from_pretrained(base_model, "ArissBandoss/mos2fr-3B")
+        except ValueError:
+            from huggingface_hub import hf_hub_download
+            adapter_config_path = hf_hub_download(
+                repo_id="ArissBarndoss/mos2fr-3B",
+                filename="adapter_config.json"
+            )
+        
+            peft_config = PeftConfig.from_json_file(adapter_config_path)
+            base_model = AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path)
+            model = PeftModel.from_pretrained(base_model, "ArissBandoss/mos2fr-3B")
+
+        
         
     trans_pipe = pipeline("translation", 
                           model=model, tokenizer=tokenizer,