barreloflube commited on
Commit
c637f12
1 Parent(s): 2646848

Refactor config.py to update MODEL_DOWNLOAD_DIR path

Browse files
Files changed (1) hide show
  1. config.py +1 -1
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', '/app/.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