Update core/image_generator.py

core/image_generator.py

@@ -117,36 +117,37 @@
 
 
 # core/image_generator.py
+# ---------------- CACHE & MODEL DIRECTORIES (FIXED) ----------------
 import os
+from pathlib import Path
+
+# Force all Hugging Face caches to /tmp/hf_cache
+HF_CACHE_DIR = Path("/tmp/hf_cache")
+HF_CACHE_DIR.mkdir(parents=True, exist_ok=True)
+
+MODEL_DIR = Path("/tmp/models/realvisxl_v4")
+MODEL_DIR.mkdir(parents=True, exist_ok=True)
+
+# MUST be set before importing diffusers/transformers
+os.environ["HF_HOME"] = str(HF_CACHE_DIR)
+os.environ["HF_HUB_CACHE"] = str(HF_CACHE_DIR)
+os.environ["DIFFUSERS_CACHE"] = str(HF_CACHE_DIR)
+os.environ["TRANSFORMERS_CACHE"] = str(HF_CACHE_DIR)
+os.environ["XDG_CACHE_HOME"] = str(HF_CACHE_DIR)
+os.environ["HF_DATASETS_CACHE"] = str(HF_CACHE_DIR)
+os.environ["HF_MODULES_CACHE"] = str(HF_CACHE_DIR)
+
+print("[DEBUG] Hugging Face cache directory set to:", HF_CACHE_DIR)
+print("[DEBUG] Model directory set to:", MODEL_DIR)
+
 import torch
 from diffusers import StableDiffusionXLPipeline
 from huggingface_hub import hf_hub_download
-from pathlib import Path
 from typing import List
 from io import BytesIO
 import base64
 from PIL import Image
 
-# ---------------- CACHE & MODEL DIRECTORIES ----------------
-HF_CACHE_DIR = Path("/tmp/hf_cache")
-MODEL_DIR = Path("/tmp/models/realvisxl_v4")
-
-# Create directories safely (no chmod)
-for d in [HF_CACHE_DIR, MODEL_DIR]:
-    d.mkdir(parents=True, exist_ok=True)
-
-# Apply environment variables BEFORE any Hugging Face usage
-os.environ.update({
-    "HF_HOME": 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),
-})
-
-# ---------------- MODEL CONFIG ----------------
-MODEL_REPO = "SG161222/RealVisXL_V4.0"
-MODEL_FILENAME = "RealVisXL_V4.0.safetensors"
 
 # ---------------- MODEL DOWNLOAD ----------------
 def download_model() -> Path: