Update app.py

app.py

@@ -33,6 +33,8 @@ def main():
 
             docsearch = FAISS.from_documents(text_chunks, embeddings)
 
+            docsearch = docsearch.as_retriever(search_type="similarity", search_kwargs={"k":1})
+
             docsearch.save_local(DB_FAISS_PATH)
 
             llm = CTransformers(model="models/llama-2-7b-chat.ggmlv3.q4_0.bin",
@@ -40,7 +42,7 @@ def main():
                                 max_new_tokens=512,
                                 temperature=0.1)
 
-            qa = ConversationalRetrievalChain.from_llm(llm, retriever=docsearch.as_retriever())
+            qa = ConversationalRetrievalChain.from_llm(llm, retriever=docsearch)
 
             result = qa(query)
             st.write(result)