Update app.py

app.py

@@ -5,7 +5,8 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
-from Gradio_UI import GradioUI
+#from Gradio_UI import GradioUI
+import gradio as gr
 
 @tool
 def execute_python_code(code: str) -> str:
@@ -131,4 +132,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+#GradioUI(agent).launch()
+gr(agent).launch()