with gr.Blocks() as demo: with gr.Row():

app.py

@@ -86,7 +86,7 @@ def get_boxes(detection_results: List[DetectionResult]) -> List[List[List[float]
 
 def mask_to_polygon(mask: np.ndarray) -> np.ndarray:
     contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-    if len(contours) == 0:
+    if len(contours) is 0:
         return np.array([])
     largest_contour = max(contours, key=cv2.contourArea)
     return largest_contour
@@ -231,7 +231,7 @@ with gr.Blocks() as demo:
 
     annotated_output = gr.Image(type="numpy")
     json_output = gr.Textbox()
-    crops_output = gr.Gallery(label="Cropped Bounding Boxes").style(grid=[2], height="auto")
+    crops_output = gr.Gallery(label="Cropped Bounding Boxes")
 
     def update_outputs(image, include_json, include_bboxes):
         results = process_image(image, include_json, include_bboxes)