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 Mistral Model  ────────────────
+# ────────────────  Load Falcon Model  ────────────────
 try:
-    pipe = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", token=hf_token)
+    pipe = pipeline("text-generation", model="tiiuae/falcon-rw-1b", token=hf_token)
 except Exception as e:
     raise RuntimeError(f"Model loading failed: {e}")
 
 # ────────────────  Agent Class  ────────────────
-class MistralAgent:
+class FalconAgent:
     def __init__(self):
-        print("✅ MistralAgent initialized.")
+        print("✅ FalconAgent initialized.")
 
     def __call__(self, question: str) -> str:
         try:
-            prompt = f"[INST] {question} [/INST]"
+            prompt = f"{question}"
             output = pipe(prompt, max_new_tokens=100, do_sample=False)
-            return output[0]["generated_text"].split("[/INST]")[-1].strip()
+            return output[0]["generated_text"].replace(prompt, "").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 = MistralAgent()
+        agent = FalconAgent()
     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("# 🧠 Mistral-7B Agent Evaluation")
+    gr.Markdown("# 🧠 Falcon-RW-1B Agent Evaluation")
     gr.Markdown(
         """
         **Instructions:**
@@ -130,5 +130,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("Launching Gradio Interface...")
-    demo.launch(debug=True, server_name="0.0.0.0", server_port=7860)
-    
+    demo.launch(debug=True, server_name="0.0.0.0", server_port=7860)