drkareemkamal commited on
Commit
a002123
·
verified ·
1 Parent(s): 79a8f20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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': 3}),
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
  )