Commit
·
c637f12
1
Parent(s):
2646848
Refactor config.py to update MODEL_DOWNLOAD_DIR path
Browse files
config.py
CHANGED
|
@@ -34,7 +34,7 @@ body {
|
|
| 34 |
class Config:
|
| 35 |
# General
|
| 36 |
SECRET_KEY = os.environ.get('SECRET_KEY', '12345678')
|
| 37 |
-
MODEL_DOWNLOAD_DIR = os.environ.get('HF_HOME', os.environ.get('HF_HUB_CACHE', '
|
| 38 |
os.makedirs(MODEL_DOWNLOAD_DIR, exist_ok=True)
|
| 39 |
|
| 40 |
# Images
|
|
|
|
| 34 |
class Config:
|
| 35 |
# General
|
| 36 |
SECRET_KEY = os.environ.get('SECRET_KEY', '12345678')
|
| 37 |
+
MODEL_DOWNLOAD_DIR = os.environ.get('HF_HOME', os.environ.get('HF_HUB_CACHE', '.cache'))
|
| 38 |
os.makedirs(MODEL_DOWNLOAD_DIR, exist_ok=True)
|
| 39 |
|
| 40 |
# Images
|