Update app.py

app.py

@@ -9,6 +9,7 @@ from smolagents.agent_types import AgentImage
 import uuid
 import requests
 import os
+from smolagents import ToolCallingAgent
 
 CRICKET_API_KEY = os.getenv("CRICKET_API_KEY")
 
@@ -193,7 +194,7 @@ with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
     
-agent = CodeAgent(
+agent = ToolCallingAgent(
     model=model,
     tools=[final_answer, generate_image, get_current_time_in_timezone, get_live_cricket_score, get_today_matches, wiki_search], ## add your tools here (don't remove final answer)
     max_steps=6,