Update app.py

app.py

@@ -12,16 +12,19 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
-    def __init__(self):
+     def __init__(self):
         print("Loading Hugging Face model...")
-        self.pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2", max_new_tokens=256)
+        self.pipeline = pipeline(
+            "text-generation",
+            model="HuggingFaceH4/zephyr-7b-beta",  # Fully public and unrestricted
+            max_new_tokens=256
+        )
         print("Model loaded.")
 
     def __call__(self, question: str) -> str:
         print(f"Received question: {question[:60]}...")
         prompt = f"Answer the question clearly and concisely:\n{question.strip()}\nAnswer:"
         response = self.pipeline(prompt)[0]["generated_text"]
-        # Strip the prompt from the output
         answer = response.split("Answer:")[-1].strip().split("\n")[0]
         print(f"Generated answer: {answer}")
         return answer