Zanqi commited on
Commit
3d11b50
·
1 Parent(s): 04da702

Update image handling in Gradio_UI to utilize gr.Image for AgentImage instances

Browse files
Files changed (1) hide show
  1. 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(