Teapack1 commited on
Commit
b02ae1e
1 Parent(s): 03c0fac

Update fast_app.py

Browse files
Files changed (1) hide show
  1. fast_app.py +1 -1
fast_app.py CHANGED
@@ -141,7 +141,7 @@ async def get_response(query: str = Form(...), language: str = Form(...)):
141
  )
142
 
143
  vectordb = FAISS.load_local(persist_directory, embedding)
144
- retriever = vectordb.as_retriever(search_kwargs={"k": 3})
145
 
146
  chain_type_kwargs = {"prompt": prompt}
147
  qa_chain = RetrievalQA.from_chain_type(
 
141
  )
142
 
143
  vectordb = FAISS.load_local(persist_directory, embedding)
144
+ retriever = vectordb.as_retriever(search_kwargs={"k": 2})
145
 
146
  chain_type_kwargs = {"prompt": prompt}
147
  qa_chain = RetrievalQA.from_chain_type(