Update inference output in app.py to yield Gradio updates instead of images, streamlining the generator's response structure for improved integration with the UI.

app.py

@@ -150,7 +150,7 @@ def infer(
         num_images_per_prompt=1,
     ).images
     stage2_only_image = stage2_images[0] if stage2_images else None
-    yield stage2_only_image, gr.update(), seed
+    yield stage2_only_image, None, seed
 
     # --- Combined generation ---
     print(f"Generating with combined LoRAs...")
@@ -180,11 +180,11 @@ def infer(
         if pil_image.size != generated_image.size:
             pil_image = pil_image.resize(generated_image.size, Image.Resampling.LANCZOS)
         blended_image = Image.blend(pil_image, generated_image, alpha=0.75)
-        yield stage2_only_image, blended_image, seed
+        yield gr.update(), blended_image, seed
         return
 
     # Return first result image and seed
-    yield stage2_only_image, result_images[0] if result_images else None, seed
+    yield gr.update(), result_images[0] if result_images else None, seed
 
 # --- Examples and UI Layout ---
 examples = []