Update core/image_generator.py

core/image_generator.py

@@ -120,6 +120,13 @@
 # ---------------- CACHE & MODEL DIRECTORIES (FIXED) ----------------
 import os
 from pathlib import Path
+import torch
+from diffusers import StableDiffusionXLPipeline
+from huggingface_hub import hf_hub_download
+from typing import List
+from io import BytesIO
+import base64
+from PIL import Image
 
 # Force all Hugging Face caches to /tmp/hf_cache
 HF_CACHE_DIR = Path("/tmp/hf_cache")
@@ -139,14 +146,11 @@ 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)
+# ---------------- MODEL CONFIG ---------------- 
+MODEL_REPO = "SG161222/RealVisXL_V4.0" 
+MODEL_FILENAME = "RealVisXL_V4.0.safetensors"
+
 
-import torch
-from diffusers import StableDiffusionXLPipeline
-from huggingface_hub import hf_hub_download
-from typing import List
-from io import BytesIO
-import base64
-from PIL import Image
 
 
 # ---------------- MODEL DOWNLOAD ----------------