Update app.py

app.py

@@ -74,12 +74,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
             num_inference_steps=steps,
             generator=torch.Generator().manual_seed(seed),
         ).images[0]
-    return image, seed, gr.Button(visible=True)
-
-@spaces.GPU
-def infer_example(input_image, prompt):
-    image, seed, _ = infer(input_image, prompt)
-    return image, seed
+    return (input_image, image), seed
 
 css="""
 #col-container {
@@ -135,8 +130,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                     )
                     
             with gr.Column():
-                result = gr.Image(label="Result", show_label=False, interactive=False)
-                reuse_button = gr.Button("Reuse this image", visible=False)
+                result = gr.ImageSlider(label="Result", show_label=False, interactive=False)
         
             
         examples = gr.Examples(
@@ -147,7 +141,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
             ],
             inputs=[input_image, prompt],
             outputs=[result, seed],
-            fn=infer_example,
+            fn=infer,
             cache_examples="lazy"
         )
             
@@ -157,10 +151,5 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
         inputs = [input_image, prompt, seed, randomize_seed, guidance_scale, steps],
         outputs = [result, seed, reuse_button]
     )
-    reuse_button.click(
-        fn = lambda image: image,
-        inputs = [result],
-        outputs = [input_image]
-    )
 
 demo.launch(mcp_server=True)