Jabrain commited on
Commit
b45225d
·
1 Parent(s): cd4675c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -197,7 +197,7 @@ if user_input:
197
  # Display the response as a chat message or an image
198
  with io_container:
199
  if output_format == "Text":
200
- st.chat_message("assistant", response)
201
  else:
202
  st.image(response, caption="Assistant output")
203
 
 
197
  # Display the response as a chat message or an image
198
  with io_container:
199
  if output_format == "Text":
200
+ st.write(f"assistant: {response}")
201
  else:
202
  st.image(response, caption="Assistant output")
203