ArturG9 commited on
Commit
852ae92
·
verified ·
1 Parent(s): 5d91633

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def retriever_from_chroma(docs, search_type, k):
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
 
40
 
 
34
  os.makedirs(vectorstore_path)
35
  vectorstore = Chroma.from_documents(
36
  documents=docs, embedding=embeddings, persist_directory="docs/chroma/")
37
+ retriever = vectorstore.as_retriever(search_type=search_type, search_kwargs={"k": k})
38
  return retriever
39
 
40