Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ logging.getLogger("chromadb.telemetry.product.posthog").setLevel(logging.CRITICA
|
|
| 50 |
|
| 51 |
# Resolve the data directory: /data when running inside HF Spaces (bucket mount),
|
| 52 |
# ./data for local development.
|
| 53 |
-
DATA_DIR = Path("/
|
| 54 |
|
| 55 |
CHROMA_PERSIST_DIR = str(DATA_DIR / "chroma_db")
|
| 56 |
TRAIN_DOCS_DIR = Path("./train_data")
|
|
|
|
| 50 |
|
| 51 |
# Resolve the data directory: /data when running inside HF Spaces (bucket mount),
|
| 52 |
# ./data for local development.
|
| 53 |
+
DATA_DIR = Path("/data_V5") if Path("/data_V5").is_dir() else Path("./data_V5")
|
| 54 |
|
| 55 |
CHROMA_PERSIST_DIR = str(DATA_DIR / "chroma_db")
|
| 56 |
TRAIN_DOCS_DIR = Path("./train_data")
|