bugfix

util.py

@@ -1,6 +1,22 @@
-# util.py (Spaces-safe + metrics)
+
+
+# util.py (patched cache handling for HF Spaces)
 import os
 from pathlib import Path
+
+# Put every cache under /tmp (always writable in Spaces)
+CACHE_DIR = os.getenv("HF_CACHE_DIR", "/tmp/hf-cache")
+Path(CACHE_DIR).mkdir(parents=True, exist_ok=True)
+
+# Make sure libraries don't fall back to "~/.cache" -> "/.cache"
+os.environ.setdefault("HF_HOME", CACHE_DIR)
+os.environ.setdefault("TRANSFORMERS_CACHE", CACHE_DIR)
+os.environ.setdefault("HUGGINGFACE_HUB_CACHE", CACHE_DIR)
+os.environ.setdefault("XDG_CACHE_HOME", CACHE_DIR)
+os.environ.setdefault("TORCH_HOME", CACHE_DIR)
+
+# util.py (Spaces-safe + metrics)
+
 from time import perf_counter
 import threading
 from io import BytesIO
@@ -11,14 +27,7 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModelForVision2Seq
 from transformers.image_utils import load_image as hf_load_image
 
-# ---- Writable caches (HF Spaces safe) ----
-CACHE_DIR = os.getenv("HF_CACHE_DIR", "/tmp/hf-cache")
-Path(CACHE_DIR).mkdir(parents=True, exist_ok=True)
-os.environ.setdefault("HF_HOME", CACHE_DIR)
-os.environ.setdefault("TRANSFORMERS_CACHE", CACHE_DIR)
-os.environ.setdefault("HUGGINGFACE_HUB_CACHE", CACHE_DIR)
-os.environ.setdefault("XDG_CACHE_HOME", CACHE_DIR)
-os.environ.setdefault("TORCH_HOME", CACHE_DIR)
+
 
 
 def _has_flash_attn() -> bool: