Update app.py

app.py

@@ -61,7 +61,7 @@ def extract_image(img, pos_prompts, neg_prompts, threshold, alpha_value=0.5):
     # convert PIL image to RGBA numpy array
     img_np = np.array(img.convert("RGBA"))
     # create an empty RGBA image with the same size
-    output_image = np.zeros_like(img_np)
+    output_image = np.zeros_like(img_np, dtype=np.uint8)
 
     # apply the final_mask as alpha channel on the output image
     output_image[:, :, :3] = img_np[:, :, :3]