undo breaking changes of width/height

app.py

@@ -108,14 +108,13 @@ with image_blocks as demo:
         with gr.Column():
             image = gr.ImageEditor(sources=["upload"], layers=False, transforms=[], 
                                    brush=gr.Brush(colors=["#000000"], color_mode="fixed"),
-                                   max_width=512, max_height=512
                                    )
             with gr.Row(elem_id="prompt-container", equal_height=True):
                 with gr.Column():  # Wrap the button inside a Column
                     btn = gr.Button("Erase!", elem_id="run_button")
         
         with gr.Column():
-            image_out = gr.Image(label="Output", elem_id="output-img", max_width=512, max_height=512)
+            image_out = gr.Image(label="Output", elem_id="output-img", height=400)
 
     # Button click will trigger the inpainting function (no prompt required)
     btn.click(fn=predict, inputs=[image], outputs=[image_out], api_name='run')