Update initialize.py

initialize.py

@@ -25,7 +25,7 @@ def initialize():
     splits = gen_splits.gen_splits()
     vectordb = Chroma.from_documents(documents=splits, persist_directory=persist_directory, embedding=embedding_model)
     vectordb.persist()
-    return collection
+    return vectordb