Update app.py

app.py

@@ -139,6 +139,7 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 
+st.button("retry", on_click=retry_last())
 
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
@@ -152,7 +153,6 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     response = get_assistant_aswer(model_links[selected_model], st.session_state.messages, temp_values, max_token_value)
     with st.chat_message("assistant"):
         st.write(response)
-        st.button("retry", on_click=retry_last())
 
     st.session_state.messages.append({"role": "assistant", "content": response})