aryswisnu commited on
Commit
8477164
·
1 Parent(s): 8b8cb9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -31,8 +31,9 @@ def process_image(image, prompt):
31
  mask_min = mask.min()
32
  mask_max = mask.max()
33
  mask = (mask - mask_min) / (mask_max - mask_min)
 
34
 
35
- ax = plt.subplots()
36
  ax.imshow(image)
37
  ax.imshow(mask, alpha=0.5, cmap="jet")
38
  ax.axis("off")
 
31
  mask_min = mask.min()
32
  mask_max = mask.max()
33
  mask = (mask - mask_min) / (mask_max - mask_min)
34
+ mask[mask < threhsold] = 0
35
 
36
+ fig, ax = plt.subplots()
37
  ax.imshow(image)
38
  ax.imshow(mask, alpha=0.5, cmap="jet")
39
  ax.axis("off")