Update app.py

app.py

@@ -356,7 +356,7 @@ if __name__ == '__main__':
         st.chat_message("user").markdown(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
 
-        ai_response = main(prompt, st.session_state.client, st.session_state.collection_name, st.session_state.llm, st.session_state.dense_model, st.session_state.sparse_model)
+        ai_response = main(prompt, client, collection_name, llm, dense_model, sparse_model)
         response = f"Echo: {ai_response}"
         with st.chat_message("assistant"):
             message_placeholder = st.empty()