Update app.py

app.py

@@ -62,12 +62,18 @@ def segment_image(input_image, text_input):
 
     for i, mask_dict in enumerate(masks):
         mask_data = (mask_dict['segmentation'] * 255).astype(np.uint8)
-        segmented_region = cv2.bitwise_and(image, image, mask=mask_data)
+
+        # Create a mask with the same shape as the original image
+        mask = np.zeros_like(image)
+        mask[:, :] = mask_data[:, :, np.newaxis]
+
+        # Apply the mask to the original image
+        segmented_region = cv2.bitwise_and(image, mask)
 
         x, y, w, h = map(int, mask_dict['bbox'])
         cropped_region = segmented_region[y:y+h, x:x+w]
 
-        # Convert the cropped region to a base64 image
+        # Convert to base64 image
         _, buffer = cv2.imencode(".png", cropped_region)
         segmented_image_base64 = base64.b64encode(buffer).decode()
 
@@ -84,7 +90,6 @@ def segment_image(input_image, text_input):
     return segmented_regions
 
 
-
 # def segment_image(input_image, text_input):
 #     image_bytes = base64.b64decode(input_image)
 #     image = Image.open(BytesIO(image_bytes))