Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from smolagents import (
-    CodeAgent,
+    ToolCallingAgent,
     OpenAIServerModel,
     DuckDuckGoSearchTool,
     WikipediaSearchTool,
@@ -40,11 +40,10 @@ class BasicAgent:
             PythonInterpreterTool(),
             get_current_date_time,
         ]
-        self.agent = CodeAgent(
+        self.agent = ToolCallingAgent(
             tools=tools,
             model=model,
             max_steps=10,
-            additional_authorized_imports=["math", "datetime", "re", "json", "collections", "itertools", "statistics"],
         )
 
     def __call__(self, question: str) -> str: