red1xe commited on
Commit
a39564e
1 Parent(s): 7991914
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ if files:
29
  chunks = text_splitter.split_text(full_text)
30
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
31
  db = FAISS.from_texts(chunks, embeddings)
32
- memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True, chat_memory=[])
33
  chain = RetrievalQA.from_llm(
34
  llm=AutoModelForCausalLM.from_pretrained("red1xe/falcon-7b-codeGPT-3K"),
35
  memory=memory,
 
29
  chunks = text_splitter.split_text(full_text)
30
  embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
31
  db = FAISS.from_texts(chunks, embeddings)
32
+ memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
33
  chain = RetrievalQA.from_llm(
34
  llm=AutoModelForCausalLM.from_pretrained("red1xe/falcon-7b-codeGPT-3K"),
35
  memory=memory,