change gr.Image() -> gr.Gallery() + remove unused kwargs

app.py

@@ -275,7 +275,7 @@ with gr.Blocks() as demo:
                 with gr.Column(scale=6, min_width=600):
                     prompt = gr.Textbox("space warrior, beautiful, female, ultrarealistic, soft lighting, 8k", placeholder="Prompt", show_label=False, lines=3)
                     negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=False, lines=3, value="3d, cartoon, anime, (deformed eyes, nose, ears, nose), bad anatomy, ugly")
-                with gr.Row(scale=6):
+                with gr.Row():
                     text_button = gr.Button("Generate", variant='primary')
                     stop_btn = gr.Button("Cancel", elem_id="cancel")
                     
@@ -323,7 +323,7 @@ with gr.Blocks() as demo:
     
                     
                 with gr.Column(scale=2):
-                    image_output = gr.Image(value="https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png")
+                    image_output = gr.Gallery(value=["https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png"], preview=True)
     
        
         with gr.Tab("img2img", id='i2i'):
@@ -331,7 +331,7 @@ with gr.Blocks() as demo:
                 with gr.Column(scale=6, min_width=600):
                     i2i_prompt = gr.Textbox("space warrior, beautiful, female, ultrarealistic, soft lighting, 8k", placeholder="Prompt", show_label=False, lines=3)
                     i2i_negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=False, lines=3, value="3d, cartoon, anime, (deformed eyes, nose, ears, nose), bad anatomy, ugly")
-                with gr.Row(scale=6):
+                with gr.Row():
                     i2i_text_button = gr.Button("Generate", variant='primary', elem_id="generate")
                     i2i_stop_btn = gr.Button("Cancel", elem_id="cancel")
                     
@@ -362,7 +362,7 @@ with gr.Blocks() as demo:
     
                     
                 with gr.Column(scale=2):
-                    i2i_image_output = gr.Image(value="https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png")
+                    i2i_image_output = gr.Gallery(value=["https://images.prodia.xyz/8ede1a7c-c0ee-4ded-987d-6ffed35fc477.png"], preview=True)
     
 
         with gr.Tab("PNG Info"):