Update app.py

app.py

@@ -17,9 +17,9 @@ import gradio as gr
 
 llm = ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key="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")
 
-if 'buffer_memory' not in st.session_state:
-    st.session_state.buffer_memory = ConversationBufferWindowMemory(k=8)
-    #st.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
+if 'buffer_memory' not in gr.session_state:
+    gr.session_state.buffer_memory = ConversationBufferWindowMemory(k=8)
+    #gr.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
 
 conversation = ConversationChain(
     llm=llm,
@@ -34,7 +34,7 @@ It is important to note that a chatbot is not a substitute for medical advice \
 from a qualified healthcare professional. \
 You respond in a short, very friendly style. \
 For each text, mark NER tags. \
-Tag categories: location, product \
+Tag categories: location, product. \
 Text: I want to visit a clinic in [Ang Mo Kio](location). \
 If a user says he/she wants to visit a traditional chinese medicine (TCM) clinic, \
 ask for which location he/she wants to visit. \
@@ -78,6 +78,7 @@ Marine Parade
 Queenstown
 Toa Payoh
 For each text, mark NER tags. \
+Tag categories: location, product. \
 Text: I want to buy/get [Po Chai Pills](product). \
 If a user wants to buy/get a product, suggest that \
 he/she can consider buying/getting from https://www.amazon.sg/s?k=product+name \