Update app.py

app.py

@@ -43,7 +43,7 @@ def extract_image(pos_prompts, neg_prompts, img, threshold):
     neg_mask = np.any(np.stack(negative_masks), axis=0)
     final_mask = pos_mask & ~neg_mask
     
-    final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
+    final_mask = Image.fromarray((final_mask * 255).astype(np.uint8), "L")
     output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
     output_image.paste(img, mask=final_mask)
     return output_image, final_mask