hopefully final path fix

app.py

@@ -17,16 +17,27 @@ import base64
 from io import BytesIO
 import tempfile
 import uuid
+import pathlib
 
-# import pathlib
-# HF_BASE = "/data/.huggingface" if os.getenv("SYSTEM") == "spaces" else "./.hf_cache" # /data/hf_cache
-# HF_BASE = str(pathlib.Path(HF_BASE).absolute())
-# os.environ.setdefault("HF_HOME", HF_BASE)
-# os.environ.setdefault("HF_HUB_CACHE", os.path.join(HF_BASE, "hub"))
-# os.environ.setdefault("HF_DATASETS_CACHE", os.path.join(HF_BASE, "datasets"))
-# os.environ.setdefault("TRANSFORMERS_CACHE", os.path.join(HF_BASE, "transformers"))
-# for k in ("HF_HUB_CACHE", "HF_DATASETS_CACHE", "TRANSFORMERS_CACHE"):
-#     os.makedirs(os.environ[k], exist_ok=True)
+# ---- Robust HF cache setup for Spaces (persistent or not) ----
+
+def _pick_hf_cache_base() -> str:
+    data = "/data"
+    if os.path.isdir(data) and os.access(data, os.W_OK):
+        return "/data/.huggingface"                 # persistent disk on Spaces
+    return "/tmp/hf_cache"                           # ephemeral but writable everywhere
+
+HF_BASE = os.environ.get("HF_HOME") or _pick_hf_cache_base()
+HF_BASE = str(pathlib.Path(HF_BASE).absolute())
+
+os.environ.setdefault("HF_HOME", HF_BASE)                           # shared root (recommended)
+os.environ.setdefault("HF_HUB_CACHE", f"{HF_BASE}/hub")             # hub cache (models/datasets/spaces)
+os.environ.setdefault("HF_DATASETS_CACHE", f"{HF_BASE}/datasets")   # datasets Arrow cache
+os.environ.setdefault("TRANSFORMERS_CACHE", f"{HF_BASE}/hub")       # transformers cache
+
+for k in ("HF_HUB_CACHE", "HF_DATASETS_CACHE", "TRANSFORMERS_CACHE"):
+    os.makedirs(os.environ[k], exist_ok=True)
+# ---------------------------------------------------------------
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = os.environ.get("CACHE_EXAMPLES") == "1"
@@ -34,7 +45,7 @@ AVAILABLE_TOKENS = list(PART_TOKENS.keys())
 
 # Download examples directly from the huggingface PartEdit-Bench
 # Login using e.g. `huggingface-cli login` or `hf login` if needed.
-bench = load_dataset("Aleksandar/PartEdit-Bench", revision="v1.1", split="synth", cache_dir=os.environ["HF_BASE"])
+bench = load_dataset("Aleksandar/PartEdit-Bench", revision="v1.1", split="synth", cache_dir=os.environ["HF_DATASETS_CACHE"])
 
 use_examples = None  # all with None
 logo = "assets/partedit.png"