Update app.py

app.py

@@ -413,10 +413,8 @@ with gr.Blocks(css="style.css") as demo:
             with gr.Row():
                 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`. """)
-                    gr.HTML("""<div id='input_image' style='display:block; margin-left: auto; margin-right: auto></div>""")
-
-                    input_image = gr.Image(source='upload', tool='sketch', type='pil', label="Upload image and draw to define mask",
-                                           width=512, height=512, brush_color='#00FFFF', mask_opacity=0.4, elem_id="input_image")    #  height=512, width=512,
+                    input_image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type='pil', label="Upload image and draw to define mask",
+                                           width=512, height=512, brush_color='#00FFFF', mask_opacity=0.4, elem_id="input_image")    
                   
                     with gr.Row():
                         sample = gr.Button("🎲 Sample New Model")