fixed cache path

app.py

@@ -18,15 +18,15 @@ from io import BytesIO
 import tempfile
 import uuid
 
-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)
+# 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)
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = os.environ.get("CACHE_EXAMPLES") == "1"
@@ -34,7 +34,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_DATASETS_CACHE"])
+bench = load_dataset("Aleksandar/PartEdit-Bench", revision="v1.1", split="synth", cache_dir=os.environ["HF_BASE"])
 
 use_examples = None  # all with None
 logo = "assets/partedit.png"