Update app.py

app.py

@@ -279,7 +279,7 @@ def use_output_as_input(output_images):
         return None
     return output_images[0]
 
-# --- Main Inference Function (with hardcoded negative prompt) ---
+    
 @spaces.GPU()
 def infer(
     reference_image,
@@ -378,6 +378,30 @@ def infer(
     # Return images, seed, and make button visible
     return image, seed, gr.update(visible=False)
 
+
+
+def infer_for_examples(
+    reference_image,
+    pose_image,
+    prompt=DEFAULT_LORA_PROMPT,
+    seed=42,
+    randomize_seed=False,
+    true_guidance_scale=1.0,
+    num_inference_steps=4,
+    height=None,
+    width=None,
+    rewrite_prompt=False,
+    num_images_per_prompt=1,
+    progress=gr.Progress(track_tqdm=True),
+):
+    """Wrapper for examples that only returns image and seed."""
+    image, seed, _ = infer(
+        reference_image, pose_image, prompt, seed, randomize_seed,
+        true_guidance_scale, num_inference_steps, height, width,
+        rewrite_prompt, num_images_per_prompt, progress
+    )
+    return image, seed
+
 # --- Examples and UI Layout ---
 examples = []
 
@@ -486,8 +510,8 @@ with gr.Blocks(css=css, theme=gr.themes.Citrus()) as demo:
             reference_image,
             pose_image,
         ],
-        outputs=[result, seed, use_output_btn],
-        fn=infer,
+        outputs=[result, seed],
+        fn=infer_for_examples,
         cache_examples=True,
         cache_mode="lazy",
     )