Spaces:

napatswift
/

table-det

Runtime error

napatswift commited on May 23, 2023

Commit

ba7e7bf

1 Parent(s): 1e41e52

Change seq type

Files changed (1) hide show

main.py CHANGED Viewed

@@ -107,7 +107,7 @@ def predict(image_input):
     for mask_image in filtered_mask_images:
         bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
-    return {'rect-fit': bbox_list, 'bbox': filtered_bboxes}
 def run():

     for mask_image in filtered_mask_images:
         bbox_list.extend(get_bbox(mask_image.astype(np.uint8)))
+    return {'rect-fit': bbox_list, 'bbox': filtered_bboxes.tolist()}
 def run():