Update agent.py

agent.py

@@ -1,4 +1,6 @@
-from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, Tool
+from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, Tool, LiteLLMModel
+
+
 
 
 class add(Tool):
@@ -134,10 +136,13 @@ tools=[
     WikipediaSearchTool()
 ]
 
+model = LiteLLMModel(
+  model_id='ollama_chat/qwen2.5:14b'
+)
 
 def create_agent():
     agent = CodeAgent(
-        model = HfApiModel(model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/'),
+        model = model,
         tools = tools,
         max_steps=10,
         verbosity_level=2