ArturG9 commited on
Commit
5d91633
·
verified ·
1 Parent(s): 2bc8ef5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def retriever_from_chroma(docs, search_type, k):
33
  if not os.path.exists(vectorstore_path):
34
  os.makedirs(vectorstore_path)
35
  vectorstore = Chroma.from_documents(
36
- documents=text_chunks, embedding=embeddings, persist_directory="docs/chroma/")
37
  retriever = vectordb.as_retriever(search_type=search_type, search_kwargs={"k": k})
38
  return retriever
39
 
 
33
  if not os.path.exists(vectorstore_path):
34
  os.makedirs(vectorstore_path)
35
  vectorstore = Chroma.from_documents(
36
+ documents=docs, embedding=embeddings, persist_directory="docs/chroma/")
37
  retriever = vectordb.as_retriever(search_type=search_type, search_kwargs={"k": k})
38
  return retriever
39