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
- highlights_api.py +5 -4
- 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 |
-
|
|
|
|
| 12 |
os.makedirs(CACHE_DIR, exist_ok=True)
|
| 13 |
-
os.makedirs(os.path.join(
|
| 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(
|
| 18 |
-
os.environ['XDG_CACHE_HOME'] = os.path.join(
|
| 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 |
-
|
|
|
|
| 13 |
os.makedirs(CACHE_DIR, exist_ok=True)
|
| 14 |
-
os.makedirs(os.path.join(
|
| 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(
|
| 19 |
-
os.environ['XDG_CACHE_HOME'] = os.path.join(
|
| 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 |
|