Spaces:
Sleeping
Sleeping
Zanqi commited on
Commit ·
3d11b50
1
Parent(s): 04da702
Update image handling in Gradio_UI to utilize gr.Image for AgentImage instances
Browse files- Gradio_UI.py +3 -1
Gradio_UI.py
CHANGED
|
@@ -164,7 +164,9 @@ def stream_to_gradio(
|
|
| 164 |
elif isinstance(final_answer, AgentImage):
|
| 165 |
yield gr.ChatMessage(
|
| 166 |
role="assistant",
|
| 167 |
-
content={"path": final_answer.to_string(), "mime_type": "image/png"},
|
|
|
|
|
|
|
| 168 |
)
|
| 169 |
elif isinstance(final_answer, AgentAudio):
|
| 170 |
yield gr.ChatMessage(
|
|
|
|
| 164 |
elif isinstance(final_answer, AgentImage):
|
| 165 |
yield gr.ChatMessage(
|
| 166 |
role="assistant",
|
| 167 |
+
# content={"path": final_answer.to_string(), "mime_type": "image/png"},
|
| 168 |
+
content=gr.Image(str(final_answer)),
|
| 169 |
+
metadata={"mime_type": "image/png"},
|
| 170 |
)
|
| 171 |
elif isinstance(final_answer, AgentAudio):
|
| 172 |
yield gr.ChatMessage(
|