Update app.py

app.py

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