Update app.py

app.py

@@ -13,22 +13,22 @@ load_dotenv()
 hf_token = os.getenv("HF_TOKEN")
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# ────────────────  Load Zephyr Model  ────────────────
+# ────────────────  Load Mistral Model  ────────────────
 try:
-    pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-alpha", token=hf_token)
+    pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", token=hf_token)
 except Exception as e:
     raise RuntimeError(f"Model loading failed: {e}")
 
 # ────────────────  Agent Class  ────────────────
-class ZephyrAgent:
+class MistralAgent:
     def __init__(self):
-        print("✅ ZephyrAgent initialized.")
+        print("✅ MistralAgent initialized.")
 
     def __call__(self, question: str) -> str:
         try:
-            prompt = f"<|system|>\nYou are a helpful agent that answers questions correctly and concisely.\n<|user|>\n{question}\n<|assistant|>"
+            prompt = f"[INST] {question} [/INST]"
             output = pipe(prompt, max_new_tokens=100, do_sample=False)
-            return output[0]["generated_text"].split("<|assistant|>")[-1].strip()
+            return output[0]["generated_text"].split("[/INST]")[-1].strip()
         except Exception as e:
             return f"LLM Error: {e}"
 
@@ -58,7 +58,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     code_link = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else ""
 
     try:
-        agent = ZephyrAgent()
+        agent = MistralAgent()
     except Exception as e:
         return f"❌ Error initializing agent: {e}", None
 
@@ -111,7 +111,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 # ────────────────  Gradio UI  ────────────────
 with gr.Blocks() as demo:
-    gr.Markdown("# 🧠 Zephyr-7B Agent Evaluation")
+    gr.Markdown("# 🧠 Mistral-7B Agent Evaluation")
     gr.Markdown(
         """
         **Instructions:**