Illia56 commited on
Commit
5b0d74e
·
1 Parent(s): bf63594

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ if "messages" not in st.session_state:
46
 
47
  # Display chat messages from history on app rerun
48
  for message in st.session_state.messages:
49
- with st.chat_message(message["role"]):
50
  st.markdown(message["content"])
51
 
52
  # React to user input
 
46
 
47
  # Display chat messages from history on app rerun
48
  for message in st.session_state.messages:
49
+ with st.chat_message(message["role"], avatar=("🧑‍💻" if message["role"]=='human' else '🌴')):
50
  st.markdown(message["content"])
51
 
52
  # React to user input