Update app.py

app.py

@@ -24,8 +24,8 @@ db = FAISS.load_local("faiss_index", HuggingFaceEmbeddings(model_name='sentence-
 
 
 retriever = db.as_retriever(
-    search_type="similarity",
-    search_kwargs={'k': 2}
+    search_type="mmr",
+    search_kwargs={'k': 1}
 )
 
 
@@ -62,7 +62,7 @@ prompt = PromptTemplate(
 llm_chain = LLMChain(llm=mistral_llm, prompt=prompt)
 
 
-retriever.search_kwargs = {'k':4}
+retriever.search_kwargs = {'k':1}
 qa = RetrievalQA.from_chain_type(
     llm=mistral_llm,
     chain_type="stuff",