Update app.py

app.py

@@ -47,7 +47,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
         height=new_height,
         generator=torch.Generator().manual_seed(seed),
     ).images[0]
-    return image, seed
+    return image, seed, gr.update(visible=True)
 
 css="""
 #col-container {
@@ -96,7 +96,7 @@ 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")
+                reuse_button = gr.Button("Reuse this image", visible=False)
         
         
 
@@ -104,7 +104,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn = infer,
         inputs = [input_image, prompt, seed, randomize_seed, guidance_scale],
-        outputs = [result, seed]
+        outputs = [result, seed, reuse_button]
     )
     reuse_button.click(
         fn = lambda image: image,