RoAr777 commited on
Commit
a3a2fed
·
verified ·
1 Parent(s): 3b80d25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -369,8 +369,7 @@ llm = HuggingFaceEndpoint(
369
  task="text-generation",
370
  timeout=10
371
  )
372
- from langchain.memory import ConversationBufferWindowMemory
373
- from langchain.memory import ChatMessageHistory
374
 
375
  llm_chain = LLMChain(prompt=prompt, llm=llm, memory=ChatMessageHistory())
376
 
 
369
  task="text-generation",
370
  timeout=10
371
  )
372
+ from langchain_community.chat_message_histories import ChatMessageHistory
 
373
 
374
  llm_chain = LLMChain(prompt=prompt, llm=llm, memory=ChatMessageHistory())
375