Update app.py

app.py

@@ -1,13 +1,18 @@
 import gradio as gr
+import numpy as np
 import imageio
 
 def dummy(img):
-  imageio.imwrite("output_image.png", img["mask"])
-  return img["image"], img["mask"]
+    imageio.imwrite("output_image.png", img["composite"])
+
+    alpha_channel = img["layers"][0][:, :, 3]
+    mask = np.where(alpha_channel == 0, 0, 255)
+
+    return img["background"], mask
 
 with gr.Blocks() as demo:
   with gr.Row():
-    img = gr.Image(tool="sketch", label="base image", show_label=True)
+    img = gr.ImageMask(sources=["upload"], layers=False, transforms=[], format="png", label="base image", show_label=True)
     img1 = gr.Image()
     img2 = gr.Image(label="mask image", show_label=True)
   btn = gr.Button()