Update app.py

app.py

@@ -40,7 +40,7 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=1500, chunk_overlap=15
 texts = text_splitter.split_documents(documents)
 embeddings = OpenAIEmbeddings()
 vectordb = Chroma.from_documents(texts,embeddings)
-chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever(),return_source_documents=True)
+chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever((search_type="mmr")),return_source_documents=True)
 
 app = flask.Flask(__name__, template_folder="./")
 # Create a directory in a known location to save files to.
@@ -108,7 +108,7 @@ def file_Upload():
         texts = text_splitter.split_documents(documents)
         embeddings = OpenAIEmbeddings()
         vectordb = Chroma.from_documents(texts,embeddings)
-        chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever(),return_source_documents=True)
+        chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever((search_type="mmr")),return_source_documents=True)
         
         return render_template("index.html")