Spaces:
Sleeping
Sleeping
Update Gradio_UI.py
Browse files- Gradio_UI.py +3 -1
Gradio_UI.py
CHANGED
@@ -157,9 +157,11 @@ def stream_to_gradio(
|
|
157 |
final_answer = handle_agent_output_types(final_answer)
|
158 |
|
159 |
if isinstance(final_answer, AgentText):
|
|
|
|
|
160 |
yield gr.ChatMessage(
|
161 |
role="assistant",
|
162 |
-
content=f"**Final answer:**\n{
|
163 |
)
|
164 |
elif isinstance(final_answer, AgentImage):
|
165 |
yield gr.ChatMessage(
|
|
|
157 |
final_answer = handle_agent_output_types(final_answer)
|
158 |
|
159 |
if isinstance(final_answer, AgentText):
|
160 |
+
formatted_final_answer = f"**Query:** {task}\n\n**Final answer:**\n{final_answer.to_string()}\n"
|
161 |
+
|
162 |
yield gr.ChatMessage(
|
163 |
role="assistant",
|
164 |
+
content=f"**Final answer:**\n{formatted_final_answer.to_string()}\n",
|
165 |
)
|
166 |
elif isinstance(final_answer, AgentImage):
|
167 |
yield gr.ChatMessage(
|