Update app.py

app.py

@@ -2,19 +2,16 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool
-import gradio as gr
+from transformers import pipeline
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Enhanced Agent Definition ---
-
 class GAIAAgent:
     def __init__(self):
         print("GAIAAgent initialized.")
-        self.model = gr.load("models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", provider="novita")
-        self.search_tool = DuckDuckGoSearchTool()  # optional if you still want to use search manually
+        self.model = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B")
 
     def format_prompt(self, question: str, file_content: str = None) -> str:
         prompt = (
@@ -36,40 +33,10 @@ 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)  # directly call the Gradio-loaded model
-        return result.strip()
-        
-# class GAIAAgent:
-#     def __init__(self):
-#         print("GAIAAgent initialized.")
-#         self.model = gr.load("models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", provider="novita")
-#         search_tool = DuckDuckGoSearchTool()
-#         self.agent = CodeAgent(model=model, tools=[search_tool])
-
-#     def format_prompt(self, question: str, file_content: str = None) -> str:
-#         prompt = (
-#             "You are a helpful AI agent solving a question from the GAIA benchmark. "
-#             "Respond only with the final answer."
-#         )
-#         if file_content:
-#             prompt += f"\nAttached File Content:\n{file_content}\n"
-#         prompt += f"\nQuestion: {question}\nAnswer:"
-#         return prompt
-
-#     def read_file(self, filename: str) -> str:
-#         filepath = os.path.join("./", filename)
-#         if filename.endswith(".txt") and os.path.exists(filepath):
-#             with open(filepath, "r") as file:
-#                 return file.read()[:1000]  # limit to 1000 chars
-#         return ""
-
-#     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.agent.run(prompt)
-#         return result.strip()
-
-        
+        result = self.model(prompt, max_new_tokens=100)[0]["generated_text"]
+        return result.strip().split("Answer:")[-1].strip()
+
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if profile:
@@ -153,4 +120,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)