Update app.py
Browse files
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")
|