Update app.py

app.py

@@ -16,8 +16,8 @@ insurers = ["Crédit Agricole","Caisse d'Epargne","Direct Assurance","Groupama",
 db_dict = {}
 
 def llm_response(insurer1, insurer2, question):
-    qa_chain1 = RetrievalQA.from_chain_type(llm1, retriever=db_dict[insurer1].as_retriever(search_kwargs={'k': 15}))
-    qa_chain2 = RetrievalQA.from_chain_type(llm2, retriever=db_dict[insurer2].as_retriever(search_kwargs={'k': 15}))
+    qa_chain1 = RetrievalQA.from_chain_type(llm1, retriever=db_dict[insurer1].as_retriever(search_kwargs={'k': 8}))
+    qa_chain2 = RetrievalQA.from_chain_type(llm2, retriever=db_dict[insurer2].as_retriever(search_kwargs={'k': 8}))
     return qa_chain1({"query": question})['result'], qa_chain2({"query": question})['result']
 
 examples = [