Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -71,8 +71,8 @@ def main():
|
|
71 |
st.write(docs)
|
72 |
#chain = load_qa_chain(llm=llm, chain_type="stuff")
|
73 |
#response = chain.run(input_documents=docs, question=query)
|
74 |
-
retriever=vector_store.as_retriever()
|
75 |
-
chain = RetrievalQA.from_chain_type(llm=llm,chain_type="stuff",retriever=retriever
|
76 |
response = chain.run(query)
|
77 |
st.write(response)
|
78 |
|
|
|
71 |
st.write(docs)
|
72 |
#chain = load_qa_chain(llm=llm, chain_type="stuff")
|
73 |
#response = chain.run(input_documents=docs, question=query)
|
74 |
+
retriever=vector_store.as_retriever(search_kargs={"k":1})
|
75 |
+
chain = RetrievalQA.from_chain_type(llm=llm,chain_type="stuff",retriever=retriever)
|
76 |
response = chain.run(query)
|
77 |
st.write(response)
|
78 |
|