Spaces:

banao-tech
/

omniapi

Sleeping

banao-tech commited on Feb 9

Commit

b7016df

verified ·

1 Parent(s): 10765ad

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -334,7 +334,7 @@ def remove_overlap_new(boxes, iou_threshold, ocr_bbox=None):
                         'content': None
                     })
             else:
-                filtered_boxes.append(box1_elem)
     return filtered_boxes  # Optionally, you could return torch.tensor(filtered_boxes) if needed

                         'content': None
                     })
             else:
+                filtered_boxes.append(box1)
     return filtered_boxes  # Optionally, you could return torch.tensor(filtered_boxes) if needed