avatar fix

app.py

@@ -96,7 +96,10 @@ bot.write("Hello human! Wanna practice naming some words?")
 #for showing history of messages
 for message in st.session_state.messages:
   with st.chat_message(message['role']):
-    st.markdown(message['content'])
+    if message['role'] == 'user':
+      st.markdown(message['content'], avatar='simon.jpg')
+    else:
+      st.markdown(message['content'])
 
 def get_text():
   input_text = st.chat_input()