Update app.py

app.py

@@ -414,7 +414,7 @@ with gr.Blocks(css="style.css") as demo:
                 with gr.Column(): 
                     gr.Markdown("""1) Either sample a new model, or upload an image (optionally draw a mask over the face) and click `invert`. """)
                     input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload image and draw to define mask",
-                                           brush_color='#00FFFF', mask_opacity=0.4)    #  height=512, width=512,
+                                           brush_color='#00FFFF', mask_opacity=0.4).style(width=512, height=512)    #  height=512, width=512,
                   
                     with gr.Row():
                         sample = gr.Button("🎲 Sample New Model")