Update app.py

app.py

@@ -270,7 +270,7 @@ def process(
             print(e)
             image = Image.new(mode="RGB", size=(512, 512))
         images.append(np.array(image))
-    return images
+    return input_image, images[0]
 
 
 #
@@ -310,7 +310,10 @@ with block:
                 latent_tiled_overlap = gr.Slider(label="Diffusion Tile Overlap", minimum=4, maximum=16, value=4, step=1)
                 scale_factor = gr.Number(label="SR Scale", value=4)
         with gr.Column():
-            result_gallery = gr.Gallery(label="Output", show_label=False, elem_id="gallery")
+            result_gallery = ImageSlider(
+                    interactive=False,
+                    label="Generated Image",
+                )
             examples = gr.Examples(
                 examples=[
                     [
@@ -323,7 +326,7 @@ with block:
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                         50,
-                        2,
+                        4,
                         7.5,
                         123,
                         320,
@@ -340,7 +343,7 @@ with block:
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                         50,
-                        2,
+                        4,
                         7.5,
                         123,
                         320,
@@ -357,7 +360,7 @@ with block:
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                         50,
-                        2,
+                        4,
                         7.5,
                         123,
                         320,