Update app.py

app.py

@@ -75,7 +75,8 @@ def get_assistant_aswer(st_model, st_messages, st_temp_value, st_max_tokens):
     return response
 
 def retry_last():
-    response = get_assistant_aswer(model_links[selected_model], st.session_state.messages[:-1], temp_values, max_token_value)
+    st.session_state.messages.pop()
+    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())