Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import inspect
 import pandas as pd
 import time
+from transformers import pipeline
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,27 +14,27 @@ 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, HF_API_TOKEN):
+    def __init__(self):
         print("BasicAgent initialized.")
-        # Initialize the model
-        #model = HfApiModel()
-        self.HF_API_TOKEN = HF_API_TOKEN
-        model = HfApiModel(model_id="openai-community/gpt2", api_key=self.HF_API_TOKEN)
+        # Initialize a small local model instead of API model
+        self.model = pipeline("text-generation", model="gpt2")
 
-        # Initialize the search tool
+        # Initialize the search tool (you can keep DuckDuckGoSearchTool)
         search_tool = DuckDuckGoSearchTool()
+
         # Initialize Agent
         self.agent = CodeAgent(
-            model = model,
+            model=self.model,
             tools=[search_tool]
         )
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer =self.agent.run(question)
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        generated_text = self.model(question, max_length=200, num_return_sequences=1)[0]['generated_text']
+        print(f"Agent returning fixed answer: {generated_text}")
+        return generated_text
 
-def run_and_submit_all( profile: gr.OAuthProfile | None, HF_API_TOKEN : str):
+def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
@@ -54,7 +55,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None, HF_API_TOKEN : str):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent(HF_API_TOKEN)
+        agent = BasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -172,9 +173,7 @@ with gr.Blocks() as demo:
     )
 
     gr.LoginButton()
-    
-    HF_API_TOKEN_box = gr.Textbox(label="HF_API_TOKEN", type="password", placeholder="sk-...", lines=1)
-    
+        
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
@@ -183,7 +182,6 @@ with gr.Blocks() as demo:
 
     run_button.click(
         fn=run_and_submit_all,
-        inputs=[HF_API_TOKEN_box],
 
 
         outputs=[status_output, results_table]