Update app.py

app.py

@@ -12,9 +12,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("Mistral Instruct Agent initialized.")
+        print("FLAN-T5 Agent initialized.")
         self.api_token = os.getenv("HF_API_TOKEN")
-        self.api_url = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
+        self.api_url = "https://api-inference.huggingface.co/models/google/flan-t5-base"
         self.headers = {
             "Authorization": f"Bearer {self.api_token}",
             "Content-Type": "application/json"
@@ -26,26 +26,35 @@ class BasicAgent:
         if not self.api_token:
             return "Error: Missing Hugging Face API token."
 
-        prompt = f"Answer this question clearly and briefly:\n{question.strip()}"
+        prompt = f"Answer this question:\n{question.strip()}"
 
         payload = {
-            "inputs": prompt
+            "inputs": prompt,
+            "parameters": {
+                "max_new_tokens": 256,
+                "temperature": 0.5
+            }
         }
 
         try:
-            response = requests.post(self.api_url, headers=self.headers, json=payload, timeout=15)
+            response = requests.post(self.api_url, headers=self.headers, json=payload, timeout=30)
             response.raise_for_status()
             output = response.json()
 
+            # Extract output
             if isinstance(output, list) and "generated_text" in output[0]:
                 return output[0]["generated_text"]
+            elif isinstance(output, dict) and "generated_text" in output:
+                return output["generated_text"]
+            elif isinstance(output, list) and "output" in output[0]:
+                return output[0]["output"]
             else:
-                print(f"Unexpected API response: {output}")
-                return "Model did not return a valid response."
+                print(f"Unexpected response: {output}")
+                return "Model returned an unexpected format."
 
         except Exception as e:
-            print(f"Exception during inference: {e}")
-            return f"API Error: {str(e)}"
+            print(f"❌ Exception during inference: {e}")
+            return f"❌ API Error: {str(e)}"
         
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):