Update app.py

app.py

@@ -43,7 +43,6 @@ def segment_image(image):
     original_size = image.size
     
     # Use higher resolution for better results while staying within model limits
-    # Most models work well with 640x640
     model_size = (640, 640)
     model_image = image.resize(model_size, Image.LANCZOS)
     
@@ -87,9 +86,12 @@ def segment_image(image):
     # Convert back to numpy
     mask_array = np.array(mask_resized) > 128
     
-    # Create visualization of mask (red on black background)
+    # Create visualization of mask (white on black background)
     mask_viz = np.zeros((mask_array.shape[0], mask_array.shape[1], 3), dtype=np.uint8)
+    # Set all channels to the same value to create white
     mask_viz[:,:,0] = mask_array * 255  # Red channel
+    mask_viz[:,:,1] = mask_array * 255  # Green channel
+    mask_viz[:,:,2] = mask_array * 255  # Blue channel
     
     return mask_array, mask_viz
 
@@ -313,7 +315,7 @@ with gr.Blocks(title="Image Blur Effects") as demo:
     gr.Markdown("4. Click 'Apply Effect'")
     gr.Markdown("")
     gr.Markdown("### Notes")
-    gr.Markdown("- The red areas in the Foreground Mask show what will remain sharp")
+    gr.Markdown("- The white areas in the Foreground Mask show what will remain sharp")
     gr.Markdown("- The Depth Map shows estimated distances (yellow=far, blue=close)")
     gr.Markdown("- Gaussian Blur applies uniform blur to the background")
     gr.Markdown("- Depth-based Blur varies blur intensity based on distance")