dharmendra commited on
Commit
58966a1
·
1 Parent(s): c1073c4

Fix: Corrected import for ConversationBufferWindowMemory

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
4
  import torch
5
  from pydantic import BaseModel
6
  import traceback
7
- from langchain.memory import ConversationBufferMemory
 
8
  from langchain.chains import ConversationChain
9
  from langchain_community.llms import HuggingFacePipeline
10
 
 
4
  import torch
5
  from pydantic import BaseModel
6
  import traceback
7
+ #from langchain.memory import ConversationBufferMemory
8
+ from langchain.memory import ConversationBufferWindowMemory
9
  from langchain.chains import ConversationChain
10
  from langchain_community.llms import HuggingFacePipeline
11