Update app.py

app.py

@@ -19,12 +19,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        # model = OpenAIServerModel(model_id="gpt-4o")
-        model = HfApiModel()
+        model = OpenAIServerModel(model_id="gpt-4o")
+        # model = HfApiModel()
         
-        self.agent = ToolCallingAgent(
+        self.agent = CodeAgent(
             tools=[DuckDuckGoSearchTool(), PythonInterpreterTool(), FinalAnswerTool()],
-            # additional_authorized_imports=['requests', 'bs4', 'csv', 'pandas', 'json', 'openpyxl', 'io', 'exec'],
+            additional_authorized_imports=['requests', 'bs4', 'csv', 'pandas', 'json', 'openpyxl', 'io', 'exec'],
             model=model,
             max_steps=10,
             verbosity_level=2