Update app.py

app.py

@@ -204,14 +204,14 @@ with image_blocks as demo:
                 with gr.Column():
                     image = gr.Image(source='upload', elem_id="image_upload",tool='sketch', type=f'{image_type}', label="Upload").style(mobile_collapse=False)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-                        btn_in = gr.Button("Erase").style(
+                        btn_in = gr.Button("Erase(↓)").style(
                             margin=True,
                             rounded=(True, True, True, True),
                             full_width=True,
                         )                
             with gr.Row():             
                 with gr.Column():
-                    image_out = gr.Image(label="Output", elem_id="image_output", visible=True)# .style(width=768)
+                    image_out = gr.Image(label="Output", elem_id="image_output", visible=True).style(width=640)
             btn_in.click(fn=predict, inputs=[image], outputs=[image_out])
             
 image_blocks.launch()