From f6a31e9563c3256b367a8511a6a8430cda5f84fb Mon Sep 17 00:00:00 2001
From: Fanghua-Yu <1901213025@pku.edu.cn>
Date: Mon, 5 Feb 2024 20:03:17 +0800
Subject: [PATCH] 20240205

---
 gradio_demo.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gradio_demo.py b/gradio_demo.py
index 87b194e..3046d96 100644
--- a/gradio_demo.py
+++ b/gradio_demo.py
@@ -143,10 +143,10 @@ with block:
             with gr.Row(equal_height=True):
                 with gr.Column():
                     gr.Markdown("<center>Input</center>")
-                    input_image = gr.Image(type="numpy", elem_id="image-input")
+                    input_image = gr.Image(type="numpy", elem_id="image-input", height=400, width=400)
                 with gr.Column():
                     gr.Markdown("<center>Stage1 Output</center>")
-                    denoise_image = gr.Image(type="numpy", elem_id="image-s1")
+                    denoise_image = gr.Image(type="numpy", elem_id="image-s1", height=400, width=400)
             prompt = gr.Textbox(label="Prompt", value="")
             with gr.Accordion("Stage1 options", open=False):
                 gamma_correction = gr.Slider(label="Gamma Correction", minimum=0.1, maximum=2.0, value=1.0, step=0.1)