new update fix

app.py

@@ -646,9 +646,13 @@ input[type="range"]::-webkit-slider-thumb {
 BASE_MODEL_ID = "black-forest-labs/FLUX.2-klein-4B"
 LORA_ID = "AnimeOverlord/flux2-klein-4b-mc"
 
-# On Hugging Face Spaces, /data is the usual persistent volume.
-MODEL_DIR = "black-forest-labs/FLUX.2-klein-4B"
-LORA_DIR = "AnimeOverlord/flux2-klein-4b-mc-v2"
+MODEL_CACHE_ROOT = os.environ.get(
+    "MODEL_CACHE_ROOT",
+    "/data/minecraftify" if os.path.isdir("/data") else ".",
+)
+MODEL_DIR = os.path.join(MODEL_CACHE_ROOT, "black-forest-labs", "FLUX.2-klein-4B")
+LORA_DIR = os.path.join(MODEL_CACHE_ROOT, "AnimeOverlord", "flux2-klein-4b-mc-v2")
+PRELOAD_MODEL_ON_STARTUP = os.environ.get("PRELOAD_MODEL_ON_STARTUP", "1") != "0"
 
 _pipe = None
 _pipe_ready = False
@@ -745,6 +749,16 @@ def _load_pipe():
     return pipe
 
 
+def preload_model_on_startup():
+    if not PRELOAD_MODEL_ON_STARTUP:
+        log_event("Startup model preload disabled by PRELOAD_MODEL_ON_STARTUP=0")
+        return
+
+    log_event("Startup model preload beginning")
+    _load_pipe()
+    log_event("Startup model preload complete")
+
+
 # ----------------------------------------------------------------------
 # INFERENCE FUNCTION
 # ----------------------------------------------------------------------
@@ -916,6 +930,7 @@ def stop_live():
 # UI
 # ----------------------------------------------------------------------
 
+preload_model_on_startup()
 log_event("Building Gradio UI")
 
 with gr.Blocks(**BLOCKS_KWARGS) as demo: