Update app.py

app.py

@@ -92,7 +92,7 @@ from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 dtype = torch.bfloat16
 
 pipe = QwenImageEditPlusPipeline.from_pretrained(
-    "FireRedTeam/FireRed-Image-Edit-1.0",
+    "FireRedTeam/FireRed-Image-Edit-1.1",
     transformer=QwenImageTransformer2DModel.from_pretrained(
         "prithivMLmods/Qwen-Image-Edit-Rapid-AIO-V19",
         torch_dtype=dtype,
@@ -109,6 +109,9 @@ except Exception as e:
 
 MAX_SEED = np.iinfo(np.int32).max
 
+DEFAULT_NEGATIVE_PROMPT = "worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
+
+
 def update_dimensions_on_upload(image):
     if image is None:
         return 1024, 1024
@@ -129,10 +132,12 @@ def update_dimensions_on_upload(image):
 
     return new_width, new_height
 
+
 @spaces.GPU
 def infer(
     images,
     prompt,
+    negative_prompt,
     seed,
     randomize_seed,
     guidance_scale,
@@ -171,7 +176,6 @@ def infer(
         seed = random.randint(0, MAX_SEED)
 
     generator = torch.Generator(device=device).manual_seed(seed)
-    negative_prompt = "worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
 
     width, height = update_dimensions_on_upload(pil_images[0])
 
@@ -189,12 +193,11 @@ def infer(
 
         return result_image, seed
 
-    except Exception as e:
-        raise e
     finally:
         gc.collect()
         torch.cuda.empty_cache()
 
+
 @spaces.GPU
 def infer_example(images, prompt):
     if not images:
@@ -208,6 +211,7 @@ def infer_example(images, prompt):
     result, seed = infer(
         images=images_list,
         prompt=prompt,
+        negative_prompt=DEFAULT_NEGATIVE_PROMPT,
         seed=0,
         randomize_seed=True,
         guidance_scale=1.0,
@@ -215,6 +219,7 @@ def infer_example(images, prompt):
     )
     return result, seed
 
+
 css = """
 #col-container {
     margin: 0 auto;
@@ -223,16 +228,24 @@ css = """
 #main-title h1 {font-size: 2.4em !important;}
 """
 
+
 with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
+
         gr.Markdown("# **FireRed-Image-Edit-1.0-Fast**", elem_id="main-title")
-        gr.Markdown("Perform image edits using [FireRed-Image-Edit-1.0](https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.0) with 4-step fast inference. Open on [GitHub](https://github.com/PRITHIVSAKTHIUR/FireRed-Image-Edit-1.0-Fast)")
+
+        gr.Markdown(
+            "Perform image edits using "
+            "[FireRed-Image-Edit-1.0](https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.0)"
+            " with 4-step fast inference."
+        )
 
         with gr.Row(equal_height=True):
+
             with gr.Column():
+
                 images = gr.Gallery(
                     label="Upload Images",
-                    #sources=["upload", "clipboard"],
                     type="filepath",
                     columns=2,
                     rows=1,
@@ -242,45 +255,70 @@ with gr.Blocks() as demo:
 
                 prompt = gr.Text(
                     label="Edit Prompt",
-                    show_label=True,
                     max_lines=2,
-                    placeholder="e.g., transform into anime, upscale, change lighting...",
+                    placeholder="e.g., transform into anime, upscale, change lighting..."
+                )
+
+                negative_prompt = gr.Textbox(
+                    label="Negative Prompt",
+                    value=DEFAULT_NEGATIVE_PROMPT,
+                    max_lines=3
                 )
 
                 run_button = gr.Button("Edit Image", variant="primary")
 
             with gr.Column():
-                output_image = gr.Image(label="Output Image", interactive=False, format="png", height=395)
+
+                output_image = gr.Image(
+                    label="Output Image",
+                    interactive=False,
+                    format="png",
+                    height=395
+                )
 
                 with gr.Accordion("Advanced Settings", open=False, visible=False):
-                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
-                    randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                    guidance_scale = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
-                    steps = gr.Slider(label="Inference Steps", minimum=1, maximum=50, step=1, value=4)
-
-        gr.Examples(
-            examples=[
-                [["examples/1.jpg"], "cinematic polaroid with soft grain subtle vignette gentle lighting white frame handwritten photographed 'Fire-Edit' preserving realistic texture and details."],
-                [["examples/2.jpg"], "Transform the image into a dotted cartoon style."],
-                [["examples/3.jpeg"], "Convert it to black and white."],
-                [["examples/4.jpg", "examples/5.jpg"], "Replace her glasses with the new glasses from image 1."],
-                [["examples/8.jpg", "examples/9.png"], "Replace the current clothing with the clothing from the reference image 2. Keep the person’s face, hairstyle, body pose, background, lighting, and camera angle unchanged. Ensure the new outfit fits naturally with realistic fabric texture, proper shadows, folds, and accurate proportions. Match the lighting, color tone, and overall style for a seamless and high-quality result."],
-                [["examples/10.jpg", "examples/11.png"], "Replace the current clothing with the clothing from the reference image 2. Keep the person’s face, hairstyle, body pose, background, lighting, and camera angle unchanged. Ensure the new outfit fits naturally with realistic fabric texture, proper shadows, folds, and accurate proportions. Match the lighting, color tone, and overall style for a seamless and high-quality result."],
-            ],
-            inputs=[images, prompt],
-            outputs=[output_image, seed],
-            fn=infer_example,
-            cache_examples=False,
-            label="Examples"
+
+                    seed = gr.Slider(
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        value=0
+                    )
+
+                    randomize_seed = gr.Checkbox(
+                        label="Randomize Seed",
+                        value=True
+                    )
+
+                    guidance_scale = gr.Slider(
+                        label="Guidance Scale",
+                        minimum=1.0,
+                        maximum=10.0,
+                        step=0.1,
+                        value=1.0
+                    )
+
+                    steps = gr.Slider(
+                        label="Inference Steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=4
+                    )
+
+        run_button.click(
+            fn=infer,
+            inputs=[images, prompt, negative_prompt, seed, randomize_seed, guidance_scale, steps],
+            outputs=[output_image, seed]
         )
-        
-        gr.Markdown("[*](https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.0)This is still an experimental Space for FireRed-Image-Edit-1.0.")
 
-    run_button.click(
-        fn=infer,
-        inputs=[images, prompt, seed, randomize_seed, guidance_scale, steps],
-        outputs=[output_image, seed]
-    )
 
 if __name__ == "__main__":
-    demo.queue(max_size=30).launch(css=css, theme=orange_red_theme, mcp_server=True, ssr_mode=False, show_error=True)
+    demo.queue(max_size=30).launch(
+        css=css,
+        theme=orange_red_theme,
+        mcp_server=True,
+        ssr_mode=False,
+        show_error=True
+    )