Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 from diffusers import StableDiffusionXLPipeline, DPMSolverMultistepScheduler
+from huggingface_hub import hf_hub_download
 import random
 
 # ── Device detection ───────────────────────────────────────────────────────────
@@ -13,16 +14,23 @@ print(f"Running on: {DEVICE.upper()} | dtype: {DTYPE}")
 MODEL_REPO = "cyberdelia/CyberRealisticPony"
 MODEL_FILE = "CyberRealisticPony_V16.0_FP16.safetensors"
 
-# ── Pony quality tags (added automatically — do not type these in your prompt) ─
+# ── Pony quality tags ──────────────────────────────────────────────────────────
 PONY_POS_PREFIX = "score_9, score_8_up, score_7_up, "
 PONY_NEG_PREFIX = "score_6, score_5, score_4, "
 
-# ── Load model ─────────────────────────────────────────────────────────────────
-print("Loading model — this may take several minutes on first start...")
+# ── Download then load from local path ────────────────────────────────────────
+print("Downloading model — this may take several minutes on first start...")
+
+local_path = hf_hub_download(
+    repo_id=MODEL_REPO,
+    filename=MODEL_FILE,
+)
+
+print(f"Model downloaded to: {local_path}")
+print("Loading pipeline...")
 
 pipe = StableDiffusionXLPipeline.from_single_file(
-    MODEL_REPO,
-    weight_name=MODEL_FILE,
+    local_path,
     torch_dtype=DTYPE,
 )
 
@@ -104,7 +112,7 @@ with gr.Blocks(title="CyberRealistic Pony", theme=gr.themes.Soft()) as demo:
                 height = gr.Slider(minimum=512, maximum=1152, value=896, step=64, label="Height")
 
             gr.Markdown("### Sampling")
-            steps    = gr.Slider(minimum=10, maximum=30, value=20, step=1,   label="Steps — keep low on CPU")
+            steps    = gr.Slider(minimum=10, maximum=30, value=20, step=1, label="Steps — keep low on CPU")
             guidance = gr.Slider(minimum=1.0, maximum=12.0, value=5.0, step=0.5, label="Guidance Scale (CFG) — recommended: 5")
 
             gr.Markdown("### Seed")