Update initialize.py

initialize.py

@@ -13,4 +13,7 @@ def initialize():
     splits = gen_splits()
     vectordb = Chroma.from_documents(documents=splits, persist_directory=persist_directory, embedding=embedding_model)
     vectordb.persist()
-return vectordb
+    return vectordb
+
+if __name__ == "__main__":
+    vectordb = initialize()