Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, VisitWebpageTool, FinalAnswerTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,13 +15,12 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         
-        # Create model using Hugging Face Inference API
-        self.model = LiteLLMModel(
-            model_id="huggingface/meta-llama/Llama-3.2-3B-Instruct",
-            api_base="https://api-inference.huggingface.co/models/",
-            max_tokens=1024
+        # Create model using Hugging Face InferenceClientModel (native to smolagents)
+        self.model = InferenceClientModel(
+            model_id="meta-llama/Llama-3.2-3B-Instruct",
+            token=None  # Will use HF_TOKEN environment variable if set
         )
-        print("Initialized Hugging Face Inference API model")
+        print("Initialized Hugging Face InferenceClientModel")
         
         # Create the CodeAgent with tools
         self.agent = CodeAgent(