Update app.py

app.py

@@ -1,4 +1,3 @@
-import numpy as np
 import streamlit as st
 from openai import OpenAI
 import os
@@ -139,8 +138,8 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 
-if st.button("retry", on_click=retry_last()):
-    pass
+if st.button("retry", key=retry):
+    retry_last()
 
 # Accept user input
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):