demoPOC commited on
Commit
394f5e8
·
1 Parent(s): bc4ba63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ text_splitter = RecursiveCharacterTextSplitter(chunk_size=1500, chunk_overlap=15
40
  texts = text_splitter.split_documents(documents)
41
  embeddings = OpenAIEmbeddings()
42
  vectordb = Chroma.from_documents(texts,embeddings)
43
- chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever((search_type="mmr")),return_source_documents=True)
44
 
45
  app = flask.Flask(__name__, template_folder="./")
46
  # Create a directory in a known location to save files to.
@@ -108,7 +108,7 @@ def file_Upload():
108
  texts = text_splitter.split_documents(documents)
109
  embeddings = OpenAIEmbeddings()
110
  vectordb = Chroma.from_documents(texts,embeddings)
111
- chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever((search_type="mmr")),return_source_documents=True)
112
 
113
  return render_template("index.html")
114
 
 
40
  texts = text_splitter.split_documents(documents)
41
  embeddings = OpenAIEmbeddings()
42
  vectordb = Chroma.from_documents(texts,embeddings)
43
+ chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever(search_type="mmr"),return_source_documents=True)
44
 
45
  app = flask.Flask(__name__, template_folder="./")
46
  # Create a directory in a known location to save files to.
 
108
  texts = text_splitter.split_documents(documents)
109
  embeddings = OpenAIEmbeddings()
110
  vectordb = Chroma.from_documents(texts,embeddings)
111
+ chain = RetrievalQA.from_chain_type(llm=OpenAI(temperature=0.0),chain_type="stuff", retriever=vectordb.as_retriever(search_type="mmr"),return_source_documents=True)
112
 
113
  return render_template("index.html")
114