Update app.py

app.py

@@ -243,7 +243,7 @@ with gr.Blocks(css=css) as demo:
                         seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    image_output = gr.Gallery(show_label=False, preview=True)
+                    image_output = gr.Image(show_label=False, type="filepath")
                     send_to_img2img = gr.Button(value="Send to img2img")
 
             
@@ -290,7 +290,7 @@ with gr.Blocks(css=css) as demo:
             i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt,
                                                    model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height,
                                                    i2i_seed], outputs=i2i_image_output, concurrency_limit=64)
-        send_to_img2img.click(send_to_img2img_def, inputs=image_output[0], outputs=i2i_image_input)
+        send_to_img2img.click(send_to_img2img_def, inputs=image_output, outputs=i2i_image_input)
         with gr.Tab("PNG Info"):
             def plaintext_to_html(text, classname=None):
                 content = "<br>\n".join(html.escape(x) for x in text.split('\n'))