uP

app.py

@@ -176,8 +176,8 @@ with gr.Blocks(css="style.css") as demo:
                         rounded=(False, True, True, False)
                     )
 
-                low_res_image = gr.Image(label="512px Image")
-                up_res_image = gr.Image(label="1024px Image")
+                low_res_image = gr.Image(label="512px Image", shape=(512, 512))
+                up_res_image = gr.Image(label="1024px Image", shape=(1024, 1024))
 
             state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(
                 container=False