Skip to content

Commit

Permalink
Updated path in config.py
Browse files Browse the repository at this point in the history
  • Loading branch information
neokd committed Oct 1, 2023
1 parent 1b78334 commit 32598bb
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions config.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
import os
from pathlib import Path
import torch
from chromadb.config import Settings
# SOURCE DOCUMENT DIRECTORY
SOURCE_DIR = os.path.join(os.path.dirname(__file__), "documents")
# To store models from HuggingFace
MODEL_DIRECTORY = os.path.join(os.path.dirname(__file__), "models")
# Store Vector Embeddings in db directory
PERSIST_DIRECTORY = os.path.join(os.path.dirname(__file__), "db")

BASE_DIR = Path(__file__).parent # Get the parent directory of this script
SOURCE_DIR = BASE_DIR / "documents" # Define the source document directory
MODEL_DIRECTORY = BASE_DIR / "models" # Define the directory to store HuggingFace models
PERSIST_DIRECTORY = BASE_DIR / "db" # Define the directory for Vector Embeddings storage


# MODELS
Expand Down

0 comments on commit 32598bb

Please sign in to comment.