fixed baby avatars

app.py

@@ -103,7 +103,7 @@ if "messages" not in st.session_state:
 for message in st.session_state.messages:
     role = message["role"]
     content = message["content"]
-    avatar = "baby.png" if role == "user" else "man.png"
+    avatar = "baby.png" if role == "assistant" else "man.png"
     with st.chat_message(role, avatar=avatar):
         st.markdown(content)