Update app.py

app.py

@@ -11,7 +11,6 @@ MODEL_FILE = "CyberRealisticPony_V16.0_FP16.safetensors"
 PONY_POS   = "score_9, score_8_up, score_7_up, "
 PONY_NEG   = "score_6, score_5, score_4, "
 
-# ── Load model on CPU at startup, ZeroGPU moves it to GPU per request ─────────
 print("Downloading model...")
 local_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 print("Loading pipeline...")
@@ -27,7 +26,7 @@ pipe.scheduler = DPMSolverMultistepScheduler.from_config(
 pipe.enable_attention_slicing()
 print("Pipeline ready.")
 
-# ── Generation — @spaces.GPU gives this function a real GPU for its duration ──
+# ── Generation ────────────────────────────────────────────────────────────────
 @spaces.GPU(duration=120)
 def generate(prompt, negative_prompt, width, height, steps, guidance, seed, randomize):
     if not prompt.strip():
@@ -36,7 +35,8 @@ def generate(prompt, negative_prompt, width, height, steps, guidance, seed, rand
         seed = random.randint(0, 2**32 - 1)
     seed = int(seed)
 
-    generator = torch.Generator(device="cuda").manual_seed(seed)
+    # CPU generator works correctly with ZeroGPU — do not hardcode "cuda"
+    generator = torch.Generator(device="cpu").manual_seed(seed)
 
     result = pipe(
         prompt=PONY_POS + prompt.strip(),
@@ -61,7 +61,6 @@ body, .gradio-container {
     margin: 0 auto !important;
 }
 
-/* Header */
 .site-header {
     background: linear-gradient(160deg, #12001a, #1e0033, #0d001a);
     border: 1px solid #6f00ff44;
@@ -109,7 +108,6 @@ body, .gradio-container {
     position: relative;
 }
 
-/* Output area */
 .output-wrap {
     background: #0f0f1a;
     border: 1px solid #1e1e35;
@@ -126,7 +124,6 @@ body, .gradio-container {
     border-radius: 18px;
 }
 
-/* Seed display */
 .seed-display input {
     background: #0f0f1a !important;
     border: 1px solid #1e1e35 !important;
@@ -137,7 +134,6 @@ body, .gradio-container {
     padding: 8px !important;
 }
 
-/* Section headers */
 .sec-head {
     color: #7733cc;
     font-size: 0.7em;
@@ -157,7 +153,6 @@ body, .gradio-container {
     background: linear-gradient(90deg, #331155, transparent);
 }
 
-/* Textareas */
 textarea {
     background: #0f0f1a !important;
     border: 1px solid #221133 !important;
@@ -173,15 +168,9 @@ textarea:focus {
     box-shadow: 0 0 0 3px #7733cc22 !important;
     outline: none !important;
 }
-textarea::placeholder {
-    color: #443355 !important;
-}
+textarea::placeholder { color: #443355 !important; }
 
-/* Sliders */
-input[type=range] {
-    accent-color: #7733cc !important;
-    height: 4px;
-}
+input[type=range] { accent-color: #7733cc !important; }
 .gradio-slider {
     background: #0f0f1a !important;
     border: 1px solid #1e1e35 !important;
@@ -197,7 +186,6 @@ input[type=range] {
     letter-spacing: 1px !important;
 }
 
-/* Number inputs */
 input[type=number] {
     background: #0f0f1a !important;
     border: 1px solid #221133 !important;
@@ -207,7 +195,6 @@ input[type=number] {
     padding: 10px !important;
 }
 
-/* Checkbox */
 .gradio-checkbox {
     background: #0f0f1a !important;
     border: 1px solid #1e1e35 !important;
@@ -225,7 +212,6 @@ input[type=checkbox] {
     font-weight: 600 !important;
 }
 
-/* Labels */
 label span {
     color: #8855bb !important;
     font-size: 0.78em !important;
@@ -234,7 +220,6 @@ label span {
     letter-spacing: 1px !important;
 }
 
-/* Generate button */
 button.lg.primary {
     background: linear-gradient(135deg, #7733cc 0%, #5500aa 50%, #3d007a 100%) !important;
     border: 1px solid #9944ee !important;
@@ -259,7 +244,6 @@ button.lg.primary:active {
     box-shadow: 0 2px 12px #7733cc44 !important;
 }
 
-/* Hide gradio footer */
 footer { display: none !important; }
 .built-with { display: none !important; }
 """
@@ -275,7 +259,6 @@ with gr.Blocks(css=css, title="CyberRealistic Pony") as demo:
     </div>
     """)
 
-    # Output — top of page so it's the first thing you see on mobile
     with gr.Group(elem_classes="output-wrap"):
         output_image = gr.Image(
             show_label=False,
@@ -284,12 +267,11 @@ with gr.Blocks(css=css, title="CyberRealistic Pony") as demo:
         )
 
     used_seed = gr.Number(
-        label="Seed used — note this down to recreate",
+        label="Seed used — note this to recreate",
         interactive=False,
         elem_classes="seed-display",
     )
 
-    # Prompt
     gr.HTML('<div class="sec-head">✏️ Prompt</div>')
     prompt = gr.Textbox(
         show_label=False,
@@ -309,18 +291,15 @@ with gr.Blocks(css=css, title="CyberRealistic Pony") as demo:
         lines=2,
     )
 
-    # Size
     gr.HTML('<div class="sec-head">📐 Size</div>')
     with gr.Row():
-        width  = gr.Slider(512, 896,  value=832, step=64, label="Width")
-        height = gr.Slider(512, 1152, value=1216, step=64, label="Height")
+        width  = gr.Slider(512, 896,  value=832,  step=64, label="Width")
+        height = gr.Slider(512, 1152, value=1024, step=64, label="Height")
 
-    # Sampling
     gr.HTML('<div class="sec-head">⚙️ Sampling</div>')
-    steps    = gr.Slider(20, 60, value=30, step=1,   label="Steps")
+    steps    = gr.Slider(20, 60,   value=30,  step=1,   label="Steps")
     guidance = gr.Slider(1.0, 12.0, value=5.0, step=0.5, label="CFG Scale")
 
-    # Seed
     gr.HTML('<div class="sec-head">🎲 Seed</div>')
     with gr.Row():
         seed           = gr.Number(label="Seed", value=42, precision=0, minimum=0, maximum=2**32-1, scale=3)