Update app.py

app.py

@@ -38,7 +38,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
         height=input_image.size[1],
         generator=torch.Generator().manual_seed(seed),
     ).images[0]
-    return image, seed, gr.update(visible=True)
+    return image, seed
 
 css="""
 #col-container {
@@ -87,9 +87,15 @@ with gr.Blocks(css=css) as demo:
                     
             with gr.Column():
                 result = gr.Image(label="Result", show_label=False, interactive=False)
-                reuse_button = gr.Button("Reuse this image", visible=False)
         
         
+    
+    gr.on(
+        triggers=[run_button.click, prompt.submit],
+        fn = infer,
+        inputs = [input_image, prompt, seed, randomize_seed, guidance_scale],
+        outputs = [result, seed]
+    )
     gr.Examples(
         examples=[
             ["./assets/5_before.png", "sunset over sea lighting coming from the top right part of the photo", 0, True, 2.5],
@@ -97,20 +103,9 @@ with gr.Blocks(css=css) as demo:
             ["./assets/2_before.png", "neon light, city",0, True, 2.5]
         ],
         inputs=[input_image, prompt, seed, randomize_seed, guidance_scale],
-        outputs=[result, seed, reuse_button],
+        outputs=[result, seed],
         fn=infer,
         cache_examples="lazy"
     )
-    gr.on(
-        triggers=[run_button.click, prompt.submit],
-        fn = infer,
-        inputs = [input_image, prompt, seed, randomize_seed, guidance_scale],
-        outputs = [result, seed, reuse_button]
-    )
-    reuse_button.click(
-        fn = lambda image: image,
-        inputs = [result],
-        outputs = [input_image]
-    )
 
 demo.launch()