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("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)
 
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):