Update app.py
Browse files
app.py
CHANGED
@@ -66,7 +66,7 @@ def extract_image(img, pos_prompts, neg_prompts, threshold, alpha_value=0.5):
|
|
66 |
output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
|
67 |
output_image.paste(img, mask=final_mask)
|
68 |
|
69 |
-
return output_image,
|
70 |
|
71 |
|
72 |
title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
|
|
|
66 |
output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
|
67 |
output_image.paste(img, mask=final_mask)
|
68 |
|
69 |
+
return output_image, mask, inverse_mask
|
70 |
|
71 |
|
72 |
title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
|