Update app.py

app.py

@@ -57,7 +57,7 @@ def detect_and_segment_persons(image, clothes):
     bboxes[:, 2] = np.clip(bboxes[:, 2], 0, width)   # x2
     bboxes[:, 3] = np.clip(bboxes[:, 3], 0, height)  # y2
 
-    segmented_images = []
+    all_segmented_images = []
     for i in range(bboxes.shape[0]):
         bbox = bboxes[i]
         x1, y1, x2, y2 = bbox
@@ -69,10 +69,10 @@ def detect_and_segment_persons(image, clothes):
         # Segment clothing for the detected person
         segmented_result = segment_clothing(pil_img, clothes)
 
-        # Append segmented results
-        segmented_images.append(segmented_result)
+        # Combine the segmented images for all persons
+        all_segmented_images.extend(segmented_result)
 
-    return segmented_images
+    return all_segmented_images
 
 @app.route('/api/detect', methods=['POST'])
 def detect():