Illia56 commited on
Commit
da2bacc
β€’
1 Parent(s): 5938dff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ if "messages" not in st.session_state:
49
 
50
  # Display chat messages from history on app rerun
51
  for message in st.session_state.messages:
52
- with st.chat_message(message["role"]):
53
  st.markdown(message["content"])
54
 
55
  # React to user input
 
49
 
50
  # Display chat messages from history on app rerun
51
  for message in st.session_state.messages:
52
+ with st.chat_message(message["role"], avatar=("πŸ§‘β€πŸ’»" if message["role"] == 'human' else 'πŸ¦™')):
53
  st.markdown(message["content"])
54
 
55
  # React to user input