rearrange avatar logic to try to fix error

app.py

@@ -216,9 +216,11 @@ with main:
     history = st.container(height=400)
     with history:
         for message in st.session_state["messages"]:
-            avatar = "🧑‍💻"
+            # avatar = "🧑‍💻"
             if message["role"] == "assistant":
                 avatar = MODEL_AVATAR_URL
+            else:
+                avatar = "🧑‍💻"
             with st.chat_message(message["role"],avatar=avatar):
                 if message["content"] is not None:
                     st.markdown(message["content"])