Fix PIL conversion

app.py

@@ -189,6 +189,7 @@ def inference(image):
 
     # Convert processed mask to PIL image
     pil_processed_mask = Image.fromarray((processed_mask * 255).astype(np.uint8)).convert('L')
+    pil_original_mask = Image.fromarray(original_mask).convert('L')
     
     im_rgb = Image.open(image).convert("RGB")
     im_dark = Image.new('RGB', im_rgb.size, (0, 0, 0))
@@ -196,7 +197,7 @@ def inference(image):
     
     # Apply processed mask to images
     im_rgba = im_rgb.copy()
-    im_rgba.putalpha(original_mask)
+    im_rgba.putalpha(pil_original_mask)
     im_dark.putalpha(pil_processed_mask)
     
     return [cropped_signature_image, processed_mask, im_dark]