Update utils/rmbg_model.py

utils/rmbg_model.py

@@ -1,10 +1,15 @@
 import os
 
-# ✅ Use a writable path inside Hugging Face Space
-CACHE_DIR = "/data/hf_cache"  # or "/tmp/hf_cache"
+# ✅ Use /tmp/hf_cache (always writable in Hugging Face Spaces)
+CACHE_DIR = "/tmp/hf_cache"
 os.environ["HF_HOME"] = CACHE_DIR
 os.environ["TRANSFORMERS_CACHE"] = CACHE_DIR
+os.environ["HF_DATASETS_CACHE"] = CACHE_DIR
+
+# Create the directory safely
 os.makedirs(CACHE_DIR, exist_ok=True)
+print("✅ Using Hugging Face cache directory:", CACHE_DIR)
+
 
 
 import torch