Update app.py

app.py

@@ -8,7 +8,7 @@ import math
 from diffusers import FlowMatchEulerDiscreteScheduler, QwenImageEditPlusPipeline
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
-from briarmbg import BriaRMBG  # ✅ background remover
+from briarmbg import BriaRMBG  
 import os
 import tempfile
 
@@ -94,7 +94,7 @@ def remove_background(image: Image.Image) -> Image.Image:
 # --- Inference ---
 @spaces.GPU
 def infer(
-    gallery_images,  # ✅ multiple images
+    gallery_images,  
     image_background,
     prompt="",
     seed=42,
@@ -113,11 +113,11 @@ def infer(
     if gallery_images:
         for gimg in gallery_images:
             pil_img = gimg[0] if isinstance(gimg, list) else gimg
-            processed_subjects.append(remove_background(pil_img))  # ✅ remove bg
+            processed_subjects.append(remove_background(pil_img)) 
 
     all_inputs = processed_subjects
     if image_background is not None:
-        all_inputs.append(image_background)  # ✅ background stays as is
+        all_inputs.append(image_background) 
 
     if not all_inputs:
         raise gr.Error("Please upload at least one image or a background image.")
@@ -141,13 +141,8 @@ css = '''#col-container { max-width: 900px; margin: 0 auto; }
 
 with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## 🧩 Qwen Image Edit — Fusion + Multi-image Input with Background Removal")
-        gr.Markdown("""
-        This version supports **multiple subject images** (auto background removal)
-        and one **background image** (kept intact).  
-        LoRAs: [Qwen Lightning](https://huggingface.co/lightx2v/Qwen-Image-Lightning) ⚡ + [Qwen Fusion](https://huggingface.co/dx8152/Qwen-Image-Edit-2509-Fusion) 🧠
-        """)
-
+        gr.Markdown("## Qwen Image Edit — Fusion")
+        gr.Markdown(""" Qwen Image Edit 2509 ✨ Using [dx8152's Qwen-Image-Edit-2509 Fusion LoRA](https://huggingface.co/dx8152/Qwen-Image-Edit-2509-Fusion) and [lightx2v Qwen-Image-Lightning LoRA]() for 4-step inference 💨 """ )
         with gr.Row():
             with gr.Column():
                 gallery = gr.Gallery(
@@ -169,17 +164,17 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
             with gr.Column():
                 result = gr.ImageSlider(label="Output Image", interactive=False)
 
-        gr.Examples(
-            examples=[
-                [["fusion_car.png", "fusion_shoes.png"], "fusion_bg.png", "put the car and shoes in the background"],
-                [["wednesday_product.png"], "simple_room.png", "put the product in her hand"]
-            ],
-            inputs=[gallery, image_background, prompt],
-            outputs=[result, seed],
-            fn=infer,
-            cache_examples="lazy",
-            elem_id="examples"
-        )
+        # gr.Examples(
+        #     examples=[
+        #         [["fusion_car.png", "fusion_shoes.png"], "fusion_bg.png", "put the car and shoes in the background"],
+        #         [["wednesday_product.png"], "simple_room.png", "put the product in her hand"]
+        #     ],
+        #     inputs=[gallery, image_background, prompt],
+        #     outputs=[result, seed],
+        #     fn=infer,
+        #     cache_examples="lazy",
+        #     elem_id="examples"
+        # )
 
         inputs = [gallery, image_background, prompt, seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width]
         outputs = [result, seed]