Spaces:
Running
Running
Add model paths and create directories for training output
Browse files- config/settings.py +10 -1
config/settings.py
CHANGED
|
@@ -18,6 +18,15 @@ load_dotenv(verbose=True)
|
|
| 18 |
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
| 19 |
VECTOR_STORE_PATH = os.path.join(BASE_DIR, "data", "vector_store")
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
# Model settings
|
| 22 |
EMBEDDING_MODEL = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
|
| 23 |
DEFAULT_MODEL = "HuggingFaceH4/zephyr-7b-beta"
|
|
@@ -28,4 +37,4 @@ if not HF_TOKEN:
|
|
| 28 |
raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
|
| 29 |
|
| 30 |
# Request settings
|
| 31 |
-
USER_AGENT = "Status-Law-Assistant/1.0"
|
|
|
|
| 18 |
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
| 19 |
VECTOR_STORE_PATH = os.path.join(BASE_DIR, "data", "vector_store")
|
| 20 |
|
| 21 |
+
# Add missing paths for training models
|
| 22 |
+
MODEL_PATH = os.path.join(BASE_DIR, "models")
|
| 23 |
+
TRAINING_OUTPUT_DIR = os.path.join(BASE_DIR, "models", "trained")
|
| 24 |
+
|
| 25 |
+
# Create directories if they don't exist
|
| 26 |
+
os.makedirs(VECTOR_STORE_PATH, exist_ok=True)
|
| 27 |
+
os.makedirs(MODEL_PATH, exist_ok=True)
|
| 28 |
+
os.makedirs(TRAINING_OUTPUT_DIR, exist_ok=True)
|
| 29 |
+
|
| 30 |
# Model settings
|
| 31 |
EMBEDDING_MODEL = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
|
| 32 |
DEFAULT_MODEL = "HuggingFaceH4/zephyr-7b-beta"
|
|
|
|
| 37 |
raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
|
| 38 |
|
| 39 |
# Request settings
|
| 40 |
+
USER_AGENT = "Status-Law-Assistant/1.0"
|