Update app.py

app.py

@@ -4,7 +4,9 @@ import datetime
 import requests
 import pytz
 import yaml
-from tools.final_answer import FinalAnswerTool
+from tools.final_answer import 
+
+from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -67,4 +69,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-agent.run("Search for the best music recommendations for a party at the Wayne's mansion.")
+GradioUI(agent).launch()