Update app.py

app.py

@@ -6,7 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
-_orig_handle = Gradio_UI.handle_agent_output_types
+_orig_handle = GradioUI.handle_agent_output_types
 
 def _patched_handle(step):
     # If step is a FinalAnswerStep, extract the inner .final_answer
@@ -15,7 +15,7 @@ def _patched_handle(step):
     # otherwise, call the original
     return _orig_handle(step)
 
-Gradio_UI.handle_agent_output_types = _patched_handle
+GradioUI.handle_agent_output_types = _patched_handle