abhivsh commited on
Commit
4f64706
·
verified ·
1 Parent(s): 12ca438

Update initialize.py

Browse files
Files changed (1) hide show
  1. initialize.py +1 -2
initialize.py CHANGED
@@ -15,8 +15,7 @@ def initialize():
15
  splits = gen_splits.gen_splits()
16
  vectordb = Chroma.from_documents(documents=splits, persist_directory=persist_directory, embedding=embedding_model)
17
  vectordb.persist()
18
- client = vectordb.PersistentClient(path=persist_directory)
19
- return vectordb, client
20
 
21
  if __name__ == "__main__":
22
  vectordb = initialize()
 
15
  splits = gen_splits.gen_splits()
16
  vectordb = Chroma.from_documents(documents=splits, persist_directory=persist_directory, embedding=embedding_model)
17
  vectordb.persist()
18
+ return vectordb
 
19
 
20
  if __name__ == "__main__":
21
  vectordb = initialize()