avinashHuggingface108 commited on
Commit
1824737
·
1 Parent(s): 77685c6

Fix: Use /tmp for cache directories to resolve permission errors

Browse files

- Changed cache directory from /app/.cache to /tmp/.cache
- /tmp is guaranteed to be writable in container environments
- Fixes PermissionError: [Errno 13] Permission denied: '/app/.cache'
- Applied to both highlights_api.py and src/smolvlm2_handler.py

Files changed (2) hide show
  1. highlights_api.py +5 -4
  2. src/smolvlm2_handler.py +5 -4
highlights_api.py CHANGED
@@ -8,14 +8,15 @@ import os
8
  import tempfile
9
 
10
  # Set cache directories to writable locations for HuggingFace Spaces
11
- CACHE_DIR = os.path.join(os.getcwd(), ".cache", "huggingface")
 
12
  os.makedirs(CACHE_DIR, exist_ok=True)
13
- os.makedirs(os.path.join(os.getcwd(), ".cache", "torch"), exist_ok=True)
14
  os.environ['HF_HOME'] = CACHE_DIR
15
  os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
16
  os.environ['HF_DATASETS_CACHE'] = CACHE_DIR
17
- os.environ['TORCH_HOME'] = os.path.join(os.getcwd(), ".cache", "torch")
18
- os.environ['XDG_CACHE_HOME'] = os.path.join(os.getcwd(), ".cache")
19
  os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
20
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
21
 
 
8
  import tempfile
9
 
10
  # Set cache directories to writable locations for HuggingFace Spaces
11
+ # Use /tmp which is guaranteed to be writable in containers
12
+ CACHE_DIR = os.path.join("/tmp", ".cache", "huggingface")
13
  os.makedirs(CACHE_DIR, exist_ok=True)
14
+ os.makedirs(os.path.join("/tmp", ".cache", "torch"), exist_ok=True)
15
  os.environ['HF_HOME'] = CACHE_DIR
16
  os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
17
  os.environ['HF_DATASETS_CACHE'] = CACHE_DIR
18
+ os.environ['TORCH_HOME'] = os.path.join("/tmp", ".cache", "torch")
19
+ os.environ['XDG_CACHE_HOME'] = os.path.join("/tmp", ".cache")
20
  os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
21
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
22
 
src/smolvlm2_handler.py CHANGED
@@ -9,14 +9,15 @@ import tempfile
9
 
10
  # Set cache directories to writable locations for HuggingFace Spaces
11
  if 'HF_HOME' not in os.environ:
12
- CACHE_DIR = os.path.join(os.getcwd(), ".cache", "huggingface")
 
13
  os.makedirs(CACHE_DIR, exist_ok=True)
14
- os.makedirs(os.path.join(os.getcwd(), ".cache", "torch"), exist_ok=True)
15
  os.environ['HF_HOME'] = CACHE_DIR
16
  os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
17
  os.environ['HF_DATASETS_CACHE'] = CACHE_DIR
18
- os.environ['TORCH_HOME'] = os.path.join(os.getcwd(), ".cache", "torch")
19
- os.environ['XDG_CACHE_HOME'] = os.path.join(os.getcwd(), ".cache")
20
  os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
21
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
22
 
 
9
 
10
  # Set cache directories to writable locations for HuggingFace Spaces
11
  if 'HF_HOME' not in os.environ:
12
+ # Use /tmp which is guaranteed to be writable in containers
13
+ CACHE_DIR = os.path.join("/tmp", ".cache", "huggingface")
14
  os.makedirs(CACHE_DIR, exist_ok=True)
15
+ os.makedirs(os.path.join("/tmp", ".cache", "torch"), exist_ok=True)
16
  os.environ['HF_HOME'] = CACHE_DIR
17
  os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
18
  os.environ['HF_DATASETS_CACHE'] = CACHE_DIR
19
+ os.environ['TORCH_HOME'] = os.path.join("/tmp", ".cache", "torch")
20
+ os.environ['XDG_CACHE_HOME'] = os.path.join("/tmp", ".cache")
21
  os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
22
  os.environ['TOKENIZERS_PARALLELISM'] = 'false'
23