Update app.py

app.py

@@ -12,16 +12,16 @@ 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="HuggingFaceH4/zephyr-7b-beta",  # Fully public and unrestricted
+            model="HuggingFaceH4/zephyr-7b-beta",
             max_new_tokens=256
         )
         print("Model loaded.")
 
-    def __call__(self, question):
+    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"]
@@ -32,6 +32,7 @@ class BasicAgent:
 
 
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,