harshitv804 commited on
Commit
9d9d0be
1 Parent(s): f187321

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ if "messages" not in st.session_state:
51
  st.session_state.messages = []
52
 
53
  if "memory" not in st.session_state:
54
- st.session_state.memory = ConversationBufferWindowMemory(k=3, memory_key="chat_history",return_messages=True)
55
 
56
  embeddings = HuggingFaceEmbeddings(model_name="BAAI/llm-embedder")
57
  db = FAISS.load_local("medchat_db", embeddings)
 
51
  st.session_state.messages = []
52
 
53
  if "memory" not in st.session_state:
54
+ st.session_state.memory = ConversationBufferWindowMemory(k=2, memory_key="chat_history",return_messages=True)
55
 
56
  embeddings = HuggingFaceEmbeddings(model_name="BAAI/llm-embedder")
57
  db = FAISS.load_local("medchat_db", embeddings)