sumit-ai-ml commited on
Commit
ea1d682
·
1 Parent(s): 235cdb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ def segment_image(input_image, selected_model):
33
  overlay_image = img.copy()
34
 
35
  # Apply the resized mask to the overlay image
36
- overlay_image[resized_mask > 0] = [50, 0, 0] # Overlay in green
37
 
38
  # Convert the overlay image to PIL format
39
  overlay_pil = Image.fromarray(overlay_image)
@@ -67,7 +67,7 @@ def segment_image(input_image, selected_model):
67
 
68
  overlay_image1 = img.copy()
69
  # Apply the resized mask to the overlay image
70
- overlay_image1[resized_mask1 > 0] = [50, 50, 0] # Overlay in green
71
 
72
  # Convert the overlay image to PIL format
73
  overlay_pil1 = Image.fromarray(overlay_image1)
 
33
  overlay_image = img.copy()
34
 
35
  # Apply the resized mask to the overlay image
36
+ overlay_image[resized_mask > 0] = [0, 255, 255] # Overlay in green
37
 
38
  # Convert the overlay image to PIL format
39
  overlay_pil = Image.fromarray(overlay_image)
 
67
 
68
  overlay_image1 = img.copy()
69
  # Apply the resized mask to the overlay image
70
+ overlay_image1[resized_mask1 > 0] = [255, 255, 0] # Overlay in green
71
 
72
  # Convert the overlay image to PIL format
73
  overlay_pil1 = Image.fromarray(overlay_image1)