Update app.py

app.py

@@ -281,20 +281,12 @@ def extract_segments(image: Image.Image, pred_seg, segment_data: Dict, settings:
             background = np.full(cropped_image.shape, 248, dtype=np.uint8)
             mask_uint8 = (cropped_mask_smooth * 255).astype(np.uint8)
             contours, _ = cv2.findContours(mask_uint8, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-            
-            outline_mask = np.zeros_like(mask_uint8)
-            cv2.drawContours(outline_mask, contours, -1, 255, thickness=2)
-            kernel = np.ones((3,3), np.uint8)
-            outline_mask = cv2.dilate(outline_mask, kernel, iterations=1)
 
             mask_3d = np.stack([cropped_mask_smooth] * 3, axis=2)
-            outline_3d = np.stack([outline_mask/255.0] * 3, axis=2)
-            grey_color = np.full(cropped_image.shape, 128, dtype=np.uint8)
-
-            # Composite image
-            final_image = (cropped_image * mask_3d +
-                          grey_color * outline_3d * (1 - mask_3d) +
-                          background * (1 - mask_3d) * (1 - outline_3d)).astype(np.uint8)
+            # Composite image - just object + white background (no grey outline)
+            final_image = (cropped_image * mask_3d + 
+              background * (1 - mask_3d)).astype(np.uint8)
+