Update app.py

app.py

@@ -1,17 +1,21 @@
 import os
 import gradio as gr
 import requests
+import inspect
 import pandas as pd
-from transformers import pipeline
+
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# --- Enhanced Agent Definition ---
 # --- Enhanced Agent Definition ---
 class GAIAAgent:
     def __init__(self):
-        print("GAIAAgent initialized.")
-        self.model = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B")
+        self.api_token = os.getenv("HF_TOKEN")
+        self.model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
+        self.api_url = f"https://api-inference.huggingface.co/models/{self.model_id}"
+        print("GAIAAgent with Hugging Face Inference API initialized.")
 
     def format_prompt(self, question: str, file_content: str = None) -> str:
         prompt = (
@@ -33,8 +37,18 @@ class GAIAAgent:
     def __call__(self, question: str, file_name: str = None) -> str:
         file_content = self.read_file(file_name) if file_name else None
         prompt = self.format_prompt(question, file_content)
-        result = self.model(prompt, max_new_tokens=100)[0]["generated_text"]
-        return result.strip().split("Answer:")[-1].strip()
+        headers = {
+            "Authorization": f"Bearer {self.api_token}",
+            "Content-Type": "application/json",
+        }
+        response = requests.post(self.api_url, headers=headers, json={"inputs": prompt})
+        result = response.json()
+
+        if isinstance(result, list) and "generated_text" in result[0]:
+            return result[0]["generated_text"].split("Answer:")[-1].strip()
+        else:
+            print(f"Unexpected response: {result}")
+            return "AGENT ERROR: Model inference failed."
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -120,4 +134,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("Launching GAIA agent app...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)