fix

inference.py

@@ -153,7 +153,7 @@ class SegmentPredictor:
             )
         sam_masks = []
         for i,mask in enumerate(masks):
-            sam_masks.append((mask["segmentation"], str(i)))
+            sam_masks.append((mask, str(i)))
         return sam_masks