Update index_query.py

index_query.py

@@ -26,7 +26,7 @@ embeddings = OpenAIEmbeddings(openai_api_key = key)
 db = FAISS.load_local("vaping_index",embeddings)
 # define retriever, score_threshold is minimum required similarity (0 to 1)
 #retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold":0.7})
-retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"k": 2,"score_threshold":0.6})
+retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"k": 3,"score_threshold":0.6})
 # create a stuff documents chain.
 sd = StuffDocumentsChain(llm_chain=llm_chain)