Update app.py

app.py

@@ -227,8 +227,8 @@ def predict(input):
 
 css = '''
 .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
-#image_upload{min-height:512px}
-#image_upload [data-testid="image"], #image_upload [data-testid="image"] > div{min-height: 512px}
+image_upload{min-height:512px}
+image_upload [data-testid="image"], #image_upload [data-testid="image"] > div{min-height: 512px}
 #mask_radio .gr-form{background:transparent; border: none}
 #word_mask{margin-top: .75em !important}
 #word_mask textarea:disabled{opacity: 0.3}
@@ -313,7 +313,7 @@ with image_blocks as demo:
         with gr.Box():
             with gr.Row():
                 with gr.Column():
-                    image = gr.Image(source='upload', elem_id='input_img', tool='sketch', type=f'{image_type}', label="Upload").style(height=auto,width=auto)
+                    image = gr.Image(source='upload', elem_id="image_upload", tool='sketch', type=f'{image_type}', label="Upload")#.style(height=auto,width=auto)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                         # prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
                         btn_in = gr.Button("Done!").style(
@@ -323,7 +323,7 @@ with image_blocks as demo:
                         )                
                 
                 # with gr.Column():
-                    image_out = gr.Image(label="Output", elem_id='output_img', visible=False) #.style(height=512)
+                    image_out = gr.Image(label="Output", elem_id="image_output", visible=False) #.style(height=512)
                     '''
                     with gr.Group(elem_id="share-btn-container"):
                         community_icon = gr.HTML(community_icon_html, visible=False)