Spaces:
Running
Running
Update utils/rmbg_model.py
Browse files- utils/rmbg_model.py +5 -6
utils/rmbg_model.py
CHANGED
|
@@ -1,11 +1,10 @@
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
-
# ✅
|
| 4 |
-
|
| 5 |
-
os.environ["
|
| 6 |
-
os.
|
| 7 |
-
|
| 8 |
-
|
| 9 |
|
| 10 |
|
| 11 |
import torch
|
|
|
|
| 1 |
import os
|
| 2 |
|
| 3 |
+
# ✅ Use a writable path inside Hugging Face Space
|
| 4 |
+
CACHE_DIR = "/data/hf_cache" # or "/tmp/hf_cache"
|
| 5 |
+
os.environ["HF_HOME"] = CACHE_DIR
|
| 6 |
+
os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
|
| 7 |
+
os.makedirs(CACHE_DIR, exist_ok=True)
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
import torch
|