improve heights

app.py

@@ -124,7 +124,7 @@ def viz_pred_mask(img, mask=None, point_coords=None, point_labels=None, bbox_coo
         for i in range(len(bbox_coords)//2):
             cv2.rectangle(out, bbox_coords[2*i], bbox_coords[2*i+1], (255,165,0), marker_size)
         if len(bbox_coords) % 2 == 1:
-            cv2.circle(out, tuple(bbox_coords[-1]), 2, (255,165,0), -1)
+            cv2.circle(out, tuple(bbox_coords[-1]), marker_size, (255,165,0), -1)
 
     return out.astype(np.uint8)
 
@@ -411,6 +411,7 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                     label="Input",
                     image_mode="L",
                     value=default_example, 
+                    container=True
                     # height=display_height
                 ) 
                 gr.Markdown("To upload your own image: click the `x` in the top right corner to clear the current image, then drag & drop")
@@ -423,7 +424,8 @@ with gr.Blocks(theme=gr.themes.Default(text_size=gr.themes.sizes.text_lg)) as de
                     elem_id="gallery",
                     preview=True, 
                     object_fit="scale-down",
-                    # height=display_height
+                    # height=display_height,
+                    container=True
                 )
     
     submit_button = gr.Button("Refresh Prediction", variant='primary')