demoPOC commited on
Commit
3e6d9da
·
verified ·
1 Parent(s): 00bd7ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,7 +42,8 @@ from langchain import PromptTemplate
42
 
43
  from langchain.chains import RetrievalQA
44
  #from langchain.memory import ConversationBufferWindowMemory
45
- from langchain.memory import ConversationBufferMemory
 
46
 
47
  from transformers import LlamaTokenizer, AutoTokenizer
48
 
 
42
 
43
  from langchain.chains import RetrievalQA
44
  #from langchain.memory import ConversationBufferWindowMemory
45
+ #from langchain.memory import ConversationBufferMemory
46
+ from langchain.memory import ConversationMemory
47
 
48
  from transformers import LlamaTokenizer, AutoTokenizer
49