invert image mask

app.py

@@ -52,11 +52,10 @@ def extract_image(img, pos_prompts, neg_prompts, threshold):
     pos_mask = np.any(np.stack(positive_masks), axis=0)
     neg_mask = np.any(np.stack(negative_masks), axis=0)
     final_mask = pos_mask & ~neg_mask
-    inverse_mask = ~pos_mask & neg_mask
 
     # extract the final image
     final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
-    inverse_mask = Image.fromarray(pos_mask.astype(np.uint8) * 255, "L")
+    inverse_mask = np.invert(final_mask)
     output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
     output_image.paste(img, mask=final_mask)