Update app.py

app.py

@@ -11,22 +11,21 @@ os.environ["STREAMLIT_HOME"] = str(CONFIG_DIR)
 os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
 os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
 
-# HF/Transformers 캐시를 **/data**로 고정 (Spaces에서 쓰기 가능)
-CACHE_ROOT = os.environ.get("HF_CACHE_ROOT", "/data")
-ENV_DIRS = {
-    "HF_HOME":               f"{CACHE_ROOT}/hf-home",
-    "TRANSFORMERS_CACHE":    f"{CACHE_ROOT}/hf-cache",
-    "HUGGINGFACE_HUB_CACHE": f"{CACHE_ROOT}/hf-cache",
-    "TORCH_HOME":            f"{CACHE_ROOT}/torch-cache",
-    "XDG_CACHE_HOME":        f"{CACHE_ROOT}/xdg-cache",
-}
-for k, v in ENV_DIRS.items():
-    os.environ[k] = v
-    os.makedirs(v, exist_ok=True)
-    try:
-        os.chmod(v, 0o777)
-    except Exception:
-        pass
+CACHE_ROOT = os.environ.get("HF_CACHE_ROOT") or str(APP_DIR / ".cache" / f"u{os.getuid()}")
+HF_HOME              = pathlib.Path(CACHE_ROOT) / "hf-home"
+TRANSFORMERS_CACHE   = pathlib.Path(CACHE_ROOT) / "hf-cache"
+HUB_CACHE            = pathlib.Path(CACHE_ROOT) / "hf-cache"
+TORCH_HOME           = pathlib.Path(CACHE_ROOT) / "torch-cache"
+XDG_CACHE_HOME       = pathlib.Path(CACHE_ROOT) / "xdg-cache"
+
+for p in [HF_HOME, TRANSFORMERS_CACHE, HUB_CACHE, TORCH_HOME, XDG_CACHE_HOME]:
+    p.mkdir(parents=True, exist_ok=True)
+
+os.environ["HF_HOME"]               = str(HF_HOME)
+os.environ["TRANSFORMERS_CACHE"]    = str(TRANSFORMERS_CACHE)
+os.environ["HUGGINGFACE_HUB_CACHE"] = str(HUB_CACHE)
+os.environ["TORCH_HOME"]            = str(TORCH_HOME)
+os.environ["XDG_CACHE_HOME"]        = str(XDG_CACHE_HOME)
 os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")