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, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool, MultiStepAgent
 from smolagents.agents import PromptTemplates
 
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
@@ -111,7 +111,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        self.agent = CodeAgent(
+        self.agent = MultiStepAgent(
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             model = model,
             add_base_tools=True,