Update app.py

app.py

@@ -173,5 +173,6 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     st.session_state.messages.append({"role": "assistant", "content": response})
 
 if len(st.session_state.messages)>0:
-    st.button("retry", key="retryButton", on_click=retry_click)
-    st.button("remove", key="removeButton", on_click=remove_click)
+    col1, col2 = st.columns(2)
+    col1.button("retry", key="retryButton", on_click=retry_click)
+    col2.button("remove", key="removeButton", on_click=remove_click)