Update app.py

app.py

@@ -3,36 +3,37 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool, HfApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, WikipediaSearchTool, InferenceClientModel
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 # --- Basic Agent Definition ---
-
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-        # Use Hugging Face Inference API instead of OpenAI
-        model = HfApiModel(model_id="meta-llama/Meta-Llama-3-8B-Instruct") #meta-llama/Llama-3.1-8B-Instruct
+        # Load the HF-supported model using InferenceClientModel
+        self.model = InferenceClientModel(
+            model_id="meta-llama/Meta-Llama-3-8B-Instruct"  #meta-llama/Llama-3.1-8B-Instruct
+        )
 
         # Tools
         duckduckgo = DuckDuckGoSearchTool()
         wikipedia = WikipediaSearchTool()
 
-        # Agent
+        # Compose the agent
         self.agent = CodeAgent(
-            model=model,
+            model=self.model,
             tools=[duckduckgo, wikipedia]
         )
 
     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
+        answer = self.agent.run(question)
+        print(f"Agent returning answer: {answer[:80]}...")
+        return answer