unit-one-introduction

Running

gperdrizet commited on Jun 20

Commit

615131f

verified ·

1 Parent(s): b6429d8

Cleaned up

Files changed (1) hide show

Gradio_UI.py CHANGED Viewed

@@ -159,7 +159,7 @@ def stream_to_gradio(
             role="assistant",
             content=str(final_answer)
         )
-    if isinstance(final_answer, AgentImage):
         yield gr.ChatMessage(
             role="assistant",
             content=gr.Image(str(final_answer)),

             role="assistant",
             content=str(final_answer)
         )
+    elif isinstance(final_answer, AgentImage):
         yield gr.ChatMessage(
             role="assistant",
             content=gr.Image(str(final_answer)),