banao-tech commited on
Commit
10765ad
·
verified ·
1 Parent(s): ab332bc

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -334,7 +334,7 @@ def remove_overlap_new(boxes, iou_threshold, ocr_bbox=None):
334
  'content': None
335
  })
336
  else:
337
- filtered_boxes.append(box1)
338
  return filtered_boxes # Optionally, you could return torch.tensor(filtered_boxes) if needed
339
 
340
 
 
334
  'content': None
335
  })
336
  else:
337
+ filtered_boxes.append(box1_elem)
338
  return filtered_boxes # Optionally, you could return torch.tensor(filtered_boxes) if needed
339
 
340