no height limit in display

app.py

@@ -90,9 +90,9 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
     justify-content: center;
 }
 #output-img img, #image_upload img {
-    object-fit: contain;
+    object-fit: contain; /* Ensure aspect ratio is preserved */
     width: 100%;
-    height: auto;
+    height: auto; /* Let height adjust automatically */
 }
 #prompt-container{margin-top:-18px;}
 #prompt-container .form{border-top-left-radius: 0;border-top-right-radius: 0}
@@ -122,7 +122,7 @@ with image_blocks as demo:
                     btn = gr.Button("Erase!", elem_id="run_button")
         
         with gr.Column():
-            image_out = gr.Image(label="Output", elem_id="output-img", height=400)
+            image_out = gr.Image(label="Output", elem_id="output-img")
 
     # Button click will trigger the inpainting function (no prompt required)
     btn.click(fn=predict, inputs=[image], outputs=[image_out], api_name='run')