Update app.py

app.py

@@ -65,12 +65,13 @@ def extract_image(img, pos_prompts, neg_prompts, threshold, alpha_value=0.5):
 
     # apply the final_mask as alpha channel on the output image
     output_image[:, :, :3] = img_np[:, :, :3]
-    output_image[:, :, 3] = (final_mask * 255 * alpha_value).astype(np.uint8)
+    output_image[:, :, 3] = (final_mask * alpha_value * 255).astype(np.uint8)
 
     # convert the output_image back to a PIL.Image object
     output_image = Image.fromarray(output_image, "RGBA")
 
-    return output_image, final_mask, bmask
+    return output_image, final_mask.astype(np.uint8), bmask
+