Add trust_remote_code=True for Mistral model compatibility

app.py

@@ -24,11 +24,15 @@ def load_model():
     peft_model_path = "yitzashapiro/FDA-guidance-zephyr-7b-beta-PEFT"
     
     try:
-        tokenizer = AutoTokenizer.from_pretrained(model_path)
+        tokenizer = AutoTokenizer.from_pretrained(
+            model_path,
+            trust_remote_code=True  # Added parameter
+        )
         model = AutoModelForCausalLM.from_pretrained(
             model_path,
             device_map="auto",
-            torch_dtype=torch.float16  # Adjust if necessary
+            torch_dtype=torch.float16,  # Adjust if necessary
+            trust_remote_code=True  # Added parameter
         ).eval()
         model.load_adapter(peft_model_path)
         st.success("Model loaded successfully.")