barreloflube
commited on
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
|