abhivsh commited on
Commit
5a0035a
·
verified ·
1 Parent(s): 7b73eee

Update initialize.py

Browse files
Files changed (1) hide show
  1. initialize.py +1 -1
initialize.py CHANGED
@@ -12,7 +12,7 @@ embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", tas
12
  persist_directory = './chroma/'
13
 
14
  def initialize():
15
- splits = gen_splits()
16
  vectordb = Chroma.from_documents(documents=splits, persist_directory=persist_directory, embedding=embedding_model)
17
  vectordb.persist()
18
  return vectordb
 
12
  persist_directory = './chroma/'
13
 
14
  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
  return vectordb