SoumyaJ commited on
Commit
7e780fe
·
verified ·
1 Parent(s): 462259e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -111,7 +111,7 @@ def loadAndSplitPdfFile(filePath):
111
 
112
  def prepare_retriever(filePath = "", load_from_chromadb = False):
113
  if load_from_chromadb:
114
- vector_store = Chroma(persist_directory=persist_directory, embedding = embeddings)
115
  return vector_store.as_retriever(search_kwargs={"k": 5})
116
  elif filePath:
117
  doc_chunks = loadAndSplitPdfFile(filePath)
 
111
 
112
  def prepare_retriever(filePath = "", load_from_chromadb = False):
113
  if load_from_chromadb:
114
+ vector_store = Chroma(persist_directory=persist_directory, embedding_function = embeddings)
115
  return vector_store.as_retriever(search_kwargs={"k": 5})
116
  elif filePath:
117
  doc_chunks = loadAndSplitPdfFile(filePath)