Upload folder using huggingface_hub

app.py

@@ -160,7 +160,6 @@ def generate_cxr(
     num_steps: int,
     cfg_scale: float,
     seed: int,
-    preview_every: int = 10,
     progress=gr.Progress(),
 ):
     """Generate a CXR, yielding intermediate previews every N steps."""
@@ -201,24 +200,6 @@ def generate_cxr(
                 progress(self._step / num_steps, desc="Generating CXR...")
                 yield item
                 self._step += 1
-                if (
-                    preview_every > 0
-                    and self._step % preview_every == 0
-                    and self._step < num_steps
-                    and "latents" in _shared_ref
-                ):
-                    try:
-                        with torch.no_grad():
-                            latents = _shared_ref["latents"]
-                            decoded = pipe.vae.decode(
-                                latents,
-                                device=pipe.device,
-                                tiled=False,
-                            )
-                            img = pipe.vae_output_to_image(decoded)
-                            previews.append(img)
-                    except Exception:
-                        pass  # skip preview on error
 
         def __len__(self):
             return len(self._iterable)
@@ -605,11 +586,10 @@ with gr.Blocks(
             # Advanced
             with gr.Accordion("Advanced Settings", open=False):
                 with gr.Row():
-                    steps_input = gr.Number(value=50, label="Steps", minimum=1, maximum=100, precision=0)
-                    cfg_input = gr.Number(value=4.0, label="CFG Scale", minimum=1.0, maximum=20.0)
+                    steps_input = gr.Number(value=40, label="Steps", minimum=1, maximum=100, precision=0)
+                    cfg_input = gr.Number(value=8.0, label="CFG Scale", minimum=1.0, maximum=20.0)
                 with gr.Row():
                     seed_input = gr.Number(value=42, label="Seed", minimum=0, precision=0)
-                    preview_every_input = gr.Number(value=10, label="Preview Every N Steps", minimum=0, maximum=50, precision=0)
 
         # ── Right column: Outputs ──
         with gr.Column(scale=2):
@@ -712,7 +692,6 @@ with gr.Blocks(
             conditions_select, severity_select,
             age_input, sex_select, view_select,
             steps_input, cfg_input, seed_input,
-            preview_every_input,
         ],
         outputs=cxr_output,
     )