Update app.py

app.py

@@ -9,7 +9,11 @@ def process_image(img):
     mask = np.where(alpha_channel == 0, 0, 255)
 
     # return img["background"], mask
-    return mask
+    # Save the mask as a PNG file
+    output_path = "mask_output.png"
+    imageio.imwrite(output_path, mask)  # Save the mask as a PNG
+
+    return output_path
 
 # with gr.Blocks() as demo:
 #   with gr.Row():