long1104 commited on
Commit
2a24833
1 Parent(s): 1d48263

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -411,7 +411,7 @@ def process_button_click(head, button_index, topic_desc, top_concept_index):
411
 
412
  st.session_state.topic_matrix[top_concept_index][button_index] += 1
413
 
414
- st.write(response)
415
  if image and caption != "": st.image(image, caption)
416
 
417
  # ... (add response & image to message)
 
411
 
412
  st.session_state.topic_matrix[top_concept_index][button_index] += 1
413
 
414
+ with st.chat_message("assistant"): st.write(response)
415
  if image and caption != "": st.image(image, caption)
416
 
417
  # ... (add response & image to message)