Update app.py
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def retrieval_qa_chain(llm,prompt,db):
|
|
69 |
qa_chain = RetrievalQA.from_chain_type(
|
70 |
llm = llm,
|
71 |
chain_type = 'stuff',
|
72 |
-
retriever = db.as_retriever(search_type = 'similarity',search_kwargs = {'k':
|
73 |
return_source_documents = True,
|
74 |
chain_type_kwargs = {'prompt': prompt}
|
75 |
)
|
|
|
69 |
qa_chain = RetrievalQA.from_chain_type(
|
70 |
llm = llm,
|
71 |
chain_type = 'stuff',
|
72 |
+
retriever = db.as_retriever(search_type = 'similarity',search_kwargs = {'k': 2}),
|
73 |
return_source_documents = True,
|
74 |
chain_type_kwargs = {'prompt': prompt}
|
75 |
)
|