Update app.py

app.py

@@ -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("remove-"+str(pos), 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("remove-"+str(pos), 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})