feat: swap Image Edit tab to Qwen-Image-Edit 2511 (50 steps, proper instruction following)

app.py

@@ -10,7 +10,7 @@ import zipfile
 from io import BytesIO
 from PIL import Image
 
-from diffusers import Flux2KleinPipeline, ZImagePipeline
+from diffusers import Flux2KleinPipeline, ZImagePipeline, QwenImageEditPlusPipeline
 from transformers import AutoProcessor, AutoModelForCausalLM
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -115,6 +115,12 @@ zimage_pipe = ZImagePipeline.from_pretrained(
 ).to(device)
 print("Z-Image Turbo loaded!")
 
+print("Loading Qwen-Image-Edit 2511...")
+qwen_pipe = QwenImageEditPlusPipeline.from_pretrained(
+    "Qwen/Qwen-Image-Edit-2511", torch_dtype=torch.bfloat16,
+).to(device)
+print("Qwen loaded!")
+
 print("Loading Florence-2 captioner...")
 florence_processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)
 florence_model = AutoModelForCausalLM.from_pretrained(
@@ -260,11 +266,10 @@ def face_swap(body_img, face_img, custom_prompt, nsfw_on, nsfw_str, swap_str,
 
 
 # ===========================================================
-# Tab 2: Image Edit
+# Tab 2: Image Edit (Qwen-Image-Edit 2511)
 # ===========================================================
-@spaces.GPU
-def image_edit(ref_images, prompt, nsfw_on, nsfw_str,
-               seed, randomize_seed, steps, guidance,
+@spaces.GPU(duration=120)
+def image_edit(ref_images, prompt, seed, randomize_seed, steps,
                progress=gr.Progress(track_tqdm=True)):
     gc.collect(); torch.cuda.empty_cache()
     try:
@@ -274,15 +279,18 @@ def image_edit(ref_images, prompt, nsfw_on, nsfw_str,
         if not prompt or not prompt.strip():
             raise gr.Error("Enter an edit prompt!")
 
-        loras = []
-        if nsfw_on:
-            loras.append(("nsfw", nsfw_str))
-        activate_loras(loras)
-
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
 
-        result = generate(images, prompt.strip(), guidance, steps, seed)
+        w, h = update_dimensions(images[0])
+        processed = [img.resize((w, h), Image.LANCZOS).convert("RGB") for img in images]
+
+        result = qwen_pipe(
+            image=processed,
+            prompt=prompt.strip(),
+            num_inference_steps=steps,
+            generator=torch.Generator(device=device).manual_seed(seed),
+        ).images[0]
         return result, seed
     finally:
         gc.collect(); torch.cuda.empty_cache()
@@ -443,18 +451,13 @@ with gr.Blocks(css=css) as demo:
 
             # ==================== IMAGE EDIT ====================
             with gr.TabItem("Image Edit"):
-                gr.Markdown("Upload image(s) and describe the edit. NSFW LoRA on by default.")
+                gr.Markdown("Powered by **Qwen-Image-Edit 2511** — instruction-based editing at 50 steps. Supports multi-reference (upload multiple images). No LoRA needed.")
                 with gr.Row():
                     with gr.Column():
-                        ie_images = gr.Gallery(label="Input Images", type="filepath", columns=2, rows=1, height=280)
+                        ie_images = gr.Gallery(label="Input Images (multi-reference supported)", type="filepath", columns=2, rows=1, height=280)
                         ie_template = gr.Dropdown(list(EDIT_TEMPLATES.keys()), value="Custom", label="Preset")
-                        ie_prompt = gr.Textbox(label="Edit prompt", lines=3, placeholder="Describe what to change...")
-                        with gr.Row():
-                            ie_nsfw = gr.Checkbox(value=True, label="NSFW LoRA")
-                            ie_nsfw_str = gr.Slider(0.0, 1.5, value=0.6, step=0.05, label="NSFW strength")
-                        with gr.Row():
-                            ie_steps = gr.Slider(1, 50, value=4, step=1, label="Steps")
-                            ie_guidance = gr.Slider(0.0, 10.0, value=1.0, step=0.1, label="Guidance")
+                        ie_prompt = gr.Textbox(label="Edit instruction", lines=3, placeholder="e.g. remove clothing, add tattoos, change background...")
+                        ie_steps = gr.Slider(10, 100, value=50, step=5, label="Steps (50 recommended)")
                         ie_seed = gr.Slider(0, MAX_SEED, value=42, step=1, label="Seed")
                         ie_rand = gr.Checkbox(value=True, label="Randomize seed")
                         ie_btn = gr.Button("Edit", variant="primary", size="lg")
@@ -464,7 +467,7 @@ with gr.Blocks(css=css) as demo:
 
                 ie_template.change(fn=lambda t: EDIT_TEMPLATES.get(t, ""), inputs=[ie_template], outputs=[ie_prompt])
                 ie_btn.click(fn=image_edit,
-                    inputs=[ie_images, ie_prompt, ie_nsfw, ie_nsfw_str, ie_seed, ie_rand, ie_steps, ie_guidance],
+                    inputs=[ie_images, ie_prompt, ie_seed, ie_rand, ie_steps],
                     outputs=[ie_out, ie_seed_out])
 
             # ==================== POSE VARIATIONS ====================