Update src/app.py

src/app.py

@@ -81,7 +81,7 @@ def main():
         st.session_state.messages.append({"role": "user", "content": prompt})
         if option == "Simple Chat":
 
-            response = generate_response(history=st.session_state.messages, query=prompt, temperature=temperature, top_k=top_k, top_p=top_p)
+            response = generate_response(history=st.session_state.messages, temperature=temperature, top_k=top_k, top_p=top_p)
             st.chat_message("assistant").markdown(response)
             # st.session_state.messages.append({"role": "assistant", "content": response})