Update app.py

app.py

@@ -87,7 +87,7 @@ for message in st.session_state.messages:
         col1.markdown(message["content"])
         col2.button("edit-"+str(pos), key="button_edit_message_"+str(pos), args=[pos], on_click=edit_message)
         if message["role"] == "assistant":
-            col2.button(str(pos), icon = ":material/thumb_up", key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
+            col2.button(str(pos), icon = ":material/delete:", key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
 
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
@@ -107,11 +107,11 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
             col1, col2 = st.columns([9,1])
             response = col1.write_stream(assistant["stream"])
             col2.button("edit-"+str(pos), key="button_edit_message_"+str(pos), args=[pos], on_click=edit_message)
-            col2.button(str(pos), icon = ":material/thumb_up", key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
+            col2.button(str(pos), icon = ":material/delete:", key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
     else:
         with st.chat_message("assistant"):
             col1, col2 = st.columns([9,1])
             response = col1.write("Failure!")
-            col2.button(str(pos), icon = ":material/thumb_up", key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
+            col2.button(str(pos), icon = ":material/delete:", key="button_remove_message_"+str(pos), args=[pos], on_click=remove_message)
 
     st.session_state.messages.append({"role": "assistant", "content": response})