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, DuckDuckGoSearchTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool , HfApiModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,14 +15,17 @@ class BasicAgent:
     def __init__(self):
         print("Initializing Smolagents-based BasicAgent...")
 
-        hf_token = os.getenv("HF_TOKEN")
-
         # Create the tools the agent can use
         self.search_tool = DuckDuckGoSearchTool()
 
+        self.model = HfApiModel(
+            model_id= "google/gemma-2b-it",
+            token=os.getenv("HF_TOKEN")
+        )
+
         # Create a reasoning agent that can call tools
         self.agent = CodeAgent(
-            model="HuggingFaceH4/zephyr-7b-alpha",  
+            model=self.model,  
             tools=[self.search_tool],
             max_steps=6,
             )