Update README.md

README.md

@@ -91,7 +91,7 @@ def detect_bboxes(masks: np.ndarray):
 pred_bbox = []
 for segment in segmentation:
     bboxes, labels = [], []
-    for ii in range(1, len(items)):
+    for ii in range(1, len(model.config.label2id)):
         mm = segment == ii
         if mm.sum() > 0:
             bbx, lab = detect_bboxes(mm.numpy())