umangchaudhry
commited on
Update app.py
Browse files
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":
|
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")
|