Update app.py

app.py

@@ -3,10 +3,10 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import DuckDuckGoSearchTool,HfApiModel,load_tool,PythonInterpreterTool,VisitWebpageTool,Tool
+from smolagents import DuckDuckGoSearchTool,HfApiModel,load_tool,PythonInterpreterTool,VisitWebpageTool,Tool, CodeAgent
 import hashlib
 import json
-from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine,CodeAgent
+from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
 
 import torch
 from tooling import ModelMathTool
@@ -91,7 +91,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
-            llm_engine=llm_engine,
+            model=llm_engine,
             tools=[web_search,python_interpreter,visit_webpage_tool], 
             max_steps=1,
             verbosity_level=1,