up

app.py

@@ -127,7 +127,7 @@ def txt_to_img(
     pipe.to("cpu")
     torch.cuda.empty_cache()
 
-    return low_res_image, up_res_image
+    return low_res_image[0], up_res_image[0]
 
 
 def replace_nsfw_images(results):
@@ -176,13 +176,8 @@ with gr.Blocks(css="style.css") as demo:
                         rounded=(False, True, True, False)
                     )
 
-                low_res_image = gr.Gallery(
-                    label="512-pix image", show_label=False, elem_id="gallery"
-                ).style(grid=[2], height=512)
-
-                up_res_image = gr.Gallery(
-                    label="1024-pix image", show_label=False, elem_id="gallery"
-                ).style(grid=[2], height=1024)
+                low_res_image = gr.Image(label="512px Image")
+                up_res_image = gr.Image(label="1024px Image")
 
             state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(
                 container=False