fix

app.py

@@ -135,10 +135,13 @@ def visualize_sam_masks(image_array):
     for i, (mask, score) in enumerate(zip(masks, scores)):
         x_offset = w * (i + 1)
         
+        # Convert mask to boolean if needed
+        mask_bool = mask.astype(bool)
+        
         # Create overlay
         overlay = image_array.copy()
         color_mask = np.zeros_like(image_array)
-        color_mask[mask] = colors[i]
+        color_mask[mask_bool] = colors[i]
         overlay = cv2.addWeighted(overlay, 0.6, color_mask, 0.4, 0)
         
         # Draw bounding box