Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from smolagents import load_tool, CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import load_tool, CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, TransformersModel
 from dotenv import load_dotenv
 
 # (Сохраните константы как есть)
@@ -33,10 +33,13 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-        self.agent = CodeAgent(tools=[search_tool],
-                               model=InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct"),
-                               planning_interval=3
-        )
+        # self.agent = CodeAgent(tools=[search_tool],
+        #                        model=InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct"),
+        #                        planning_interval=3
+        # )
+
+        self.model = TransformersModel(model_id="HuggingFaceTB/SmolLM2-1.7B-Instruct", device_map="auto", max_new_tokens=1000)
+        self.agent = ToolCallingAgent(tools=[search_tool], model=model, verbosity_level=2)
         
         print("Agent initialized successfully.")