Update app.py

app.py

@@ -169,14 +169,14 @@ with gr.Blocks(css="style.css") as demo:
             with gr.Group(elem_id="input_image_container", elem_classes="image_container" ):
                 image = gr.Image(
                     source="canvas", tool="color-sketch", type="pil", image_mode="L",
-                    invert_colors=True, shape=(512, 512), brush_radius=4, 
+                    invert_colors=True, shape=(512, 512), brush_radius=4, height=440, width=440,
                     brush_color="#000000", interactive=True, show_download_button=False, elem_id="input_image", show_label=False)
                 gr.HTML("""<img src="assets/drawCta.png" id="draw_cta" alt="draw here image" />
                     <button id="eraser" onclick="return DELETE_SKETCH_FUNCTION(this)">
                     <span id="eraser_icon"></span>
                     </button>""",elem_id="eraser_container")
             with gr.Group(elem_id="output_image_container", elem_classes="image_container"):
-                result = gr.Image(label="Result", elem_id="output_image", show_label=False, show_download_button=True)
+                result = gr.Image(label="Result", height=440, width=440, elem_id="output_image", show_label=False, show_download_button=True)
         with gr.Row(elem_id="radio_row"):
             item = gr.Radio(choices=ITEMS_NAMES, value=DEFAULT_ITEM_NAME, show_label=False)
     gr.HTML("""<div id="footer_block">