Update app.py

app.py

@@ -76,6 +76,7 @@ MAX_SEED = np.iinfo(np.int32).max
 @spaces.GPU
 def infer(
     image_subject,image_background,
+    prompt,
     seed=42,
     randomize_seed=True,
     true_guidance_scale=1,
@@ -85,13 +86,12 @@ def infer(
     progress=gr.Progress(track_tqdm=True)
 ):
     
-    prompt=""
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device=device).manual_seed(seed)
 
     result = pipe(
-        image=image_subject,
+        image=image_subject["composite"],
         prompt=prompt,
         # height=height,
         # width=width,
@@ -117,16 +117,16 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
         gr.Markdown("## Qwen Image Edit — Fusion")
         gr.Markdown("""
             Qwen Image Edit 2509 ✨ 
-            Using [Alissonerdx's Qwen-Edit-2509 Face Swap LoRA]() and [Phr00t/Qwen-Image-Edit-Rapid-AIO]() for 4-step inference 💨
+            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():
                 with gr.Row():
-                    image_subject = gr.Image(label="subject Image", type="pil")
-                    image_background = gr.Image(label="background Image", type="pil")
-            
+                    image_subject = gr.ImageEditor(label="input image", type="pil")
+                    image_background = gr.Image(label="background Image", type="pil", visible=False)
+                prompt =  gr.Textbox(label="prompt")
                 run_button = gr.Button("Fuse", variant="primary")
                 with gr.Accordion("Advanced Settings", open=False):
                         seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
@@ -142,20 +142,20 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 result = gr.Image(label="Output Image", interactive=False)
                 prompt_preview = gr.Textbox(label="Processed Prompt", interactive=False, visible=False)
 
-        gr.Examples(
-        examples=[
-            ["wednesday.png", "pexels-alipazani-2613260.jpg"],
-
-        ],
-        inputs=[image_subject,image_background],
-        outputs=[result,seed],
-        fn=infer,
-        cache_examples="lazy",
-        elem_id="examples"
-    )
+    #     gr.Examples(
+    #     examples=[
+    #         ["wednesday.png", "pexels-alipazani-2613260.jpg"],
+
+    #     ],
+    #     inputs=[image_subject,image_background],
+    #     outputs=[result,seed],
+    #     fn=infer,
+    #     cache_examples="lazy",
+    #     elem_id="examples"
+    # )
                     
     inputs = [
-        image_subject,image_background,
+        image_subject,image_background, prompt,
         seed, randomize_seed, true_guidance_scale, num_inference_steps, height, width
     ]
     outputs = [result, seed]