Update api/server.py

api/server.py

@@ -3,6 +3,38 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from services import queue_manager
+import os
+from pathlib import Path
+
+# CACHE PATCH BLOCK: place FIRST in pipeline.py!
+HF_CACHE_DIR = Path("/tmp/hf_cache")
+HF_CACHE_DIR.mkdir(parents=True, exist_ok=True)
+os.environ.update({
+    "HF_HOME": str(HF_CACHE_DIR),
+    "HF_HUB_CACHE": str(HF_CACHE_DIR),
+    "DIFFUSERS_CACHE": str(HF_CACHE_DIR),
+    "TRANSFORMERS_CACHE": str(HF_CACHE_DIR),
+    "XDG_CACHE_HOME": str(HF_CACHE_DIR),
+    "HF_DATASETS_CACHE": str(HF_CACHE_DIR),
+    "HF_MODULES_CACHE": str(HF_CACHE_DIR),
+    "TMPDIR": str(HF_CACHE_DIR),
+    "CACHE_DIR": str(HF_CACHE_DIR),
+    "TORCH_HOME": str(HF_CACHE_DIR),
+    "HOME": str(HF_CACHE_DIR)
+})
+import os.path
+if not hasattr(os.path, "expanduser_original"):
+    os.path.expanduser_original = os.path.expanduser
+def safe_expanduser(path):
+    if (
+        path.startswith("~") or 
+        path.startswith("/.cache") or 
+        path.startswith("/root/.cache")
+    ):
+        return str(HF_CACHE_DIR)
+    return os.path.expanduser_original(path)
+os.path.expanduser = safe_expanduser
+
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")