umangchaudhry commited on
Commit
52e78c0
·
verified ·
1 Parent(s): d44a8b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def multi_plan_qa_multi_vectorstore(api_key, input_text, display_placeholder):
139
  vector_store = FAISS.load_local(vectorstore_path, embeddings, allow_dangerous_deserialization=True)
140
 
141
  # Convert the vector store to a retriever
142
- retriever = vector_store.as_retriever(search_kwargs={"k": 10})
143
 
144
  # Retrieve relevant chunks for the input text
145
  retrieved_chunks = retriever.invoke("input_text")
 
139
  vector_store = FAISS.load_local(vectorstore_path, embeddings, allow_dangerous_deserialization=True)
140
 
141
  # Convert the vector store to a retriever
142
+ retriever = vector_store.as_retriever(search_kwargs={"k": 4})
143
 
144
  # Retrieve relevant chunks for the input text
145
  retrieved_chunks = retriever.invoke("input_text")