MaryamKarimi080 commited on
Commit
0ee80e8
·
verified ·
1 Parent(s): 041e514

Update scripts/setup_vectorstore.py

Browse files
Files changed (1) hide show
  1. scripts/setup_vectorstore.py +2 -0
scripts/setup_vectorstore.py CHANGED
@@ -7,6 +7,8 @@ def main():
7
  BASE_DIR = Path(__file__).resolve().parent.parent
8
  CHUNKS_PATH = BASE_DIR / "output" / "chunks.pkl"
9
  DB_DIR = BASE_DIR / "db"
 
 
10
 
11
  with open(CHUNKS_PATH, "rb") as f:
12
  chunks = pickle.load(f)
 
7
  BASE_DIR = Path(__file__).resolve().parent.parent
8
  CHUNKS_PATH = BASE_DIR / "output" / "chunks.pkl"
9
  DB_DIR = BASE_DIR / "db"
10
+
11
+ DB_DIR.mkdir(parents=True, exist_ok=True) # <-- create the db folder if not exists
12
 
13
  with open(CHUNKS_PATH, "rb") as f:
14
  chunks = pickle.load(f)