nirmalaag commited on
Commit
bcdac7f
·
verified ·
1 Parent(s): bec5cdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ def main():
71
  chain = load_summarize_chain(llm, chain_type="map_reduce")
72
  response = chain.run(chunks_doc[0])
73
  else:
74
- docs = vector_store.similarity_search(query=query, k=1)
75
  st.write(docs)
76
  chain = load_qa_chain(llm=llm, chain_type="stuff")
77
  response = chain.run(input_documents=docs, question=query)
 
71
  chain = load_summarize_chain(llm, chain_type="map_reduce")
72
  response = chain.run(chunks_doc[0])
73
  else:
74
+ docs = vector_store.similarity_search(query=query, k=2)
75
  st.write(docs)
76
  chain = load_qa_chain(llm=llm, chain_type="stuff")
77
  response = chain.run(input_documents=docs, question=query)