Update app.py

app.py

@@ -190,7 +190,6 @@ def infer(
     true_guidance_scale=1.0,
     num_inference_steps=8,  # Default to 8 steps for fast inference
     rewrite_prompt=True,
-    num_images_per_prompt=1,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -213,7 +212,7 @@ def infer(
         prompt = polish_prompt(prompt, image)
         print(f"Rewritten Prompt: {prompt}")
 
-    # Generate the edited image
+    # Generate the edited image - always generate just 1 image
     try:
         images = pipe(
             image,
@@ -222,14 +221,16 @@ def infer(
             num_inference_steps=num_inference_steps,
             generator=generator,
             true_cfg_scale=true_guidance_scale,
-            num_images_per_prompt=num_images_per_prompt
+            num_images_per_prompt=1  # Always generate only 1 image
         ).images
+        
+        # Return the first (and only) image
+        return images[0], seed
+        
     except Exception as e:
         print(f"Error during inference: {e}")
         raise e
 
-    return images, seed
-
 # --- Examples and UI Layout ---
 examples = [
     # You can add example pairs of [image_path, prompt] here
@@ -256,7 +257,7 @@ with gr.Blocks(css=css) as demo:
         gr.HTML("""
         <div id="logo-title">
             <img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_edit_logo.png" alt="Qwen-Image Edit Logo" width="400" style="display: block; margin: 0 auto;">
-            <h2 style="font-style: italic;color: #5b47d1;margin-top: -33px !important;margin-left: 133px;">Fast, 8-steps with Lightning LoRA</h2>
+            <h2 style="font-style: italic;color: #5b47d1;margin-top: -27px !important;margin-left: 133px;">Fast, 8-steps with Lightning LoRA</h2>
         </div>
         """)
         gr.Markdown("""
@@ -272,7 +273,8 @@ with gr.Blocks(css=css) as demo:
                     show_label=True, 
                     type="pil"
                 )
-            result = gr.Gallery(
+            # Changed from Gallery to Image
+            result = gr.Image(
                 label="Result", 
                 show_label=True, 
                 type="pil"
@@ -315,20 +317,11 @@ with gr.Blocks(css=css) as demo:
                     value=8
                 )
                 
-            with gr.Row():
-                num_images_per_prompt = gr.Slider(
-                    label="Number of images per prompt",
-                    minimum=1,
-                    maximum=4,
-                    step=1,
-                    value=1,
-                    visible=False
-                )
-                
-                rewrite_prompt = gr.Checkbox(
-                    label="Enhance prompt (using HF Inference)", 
-                    value=True
-                )
+            # Removed num_images_per_prompt slider entirely
+            rewrite_prompt = gr.Checkbox(
+                label="Enhance prompt (using HF Inference)", 
+                value=True
+            )
 
         # gr.Examples(examples=examples, inputs=[input_image, prompt], outputs=[result, seed], fn=infer, cache_examples=False)
 
@@ -343,7 +336,7 @@ with gr.Blocks(css=css) as demo:
             true_guidance_scale,
             num_inference_steps,
             rewrite_prompt,
-            num_images_per_prompt,
+            # Removed num_images_per_prompt from inputs
         ],
         outputs=[result, seed],
     )