Update app.py

app.py

@@ -8,7 +8,8 @@ def process_image(img):
     alpha_channel = img["layers"][0][:, :, 3]
     mask = np.where(alpha_channel == 0, 0, 255)
 
-    return img["background"], mask
+    # return img["background"], mask
+    return mask
 
 # with gr.Blocks() as demo:
 #   with gr.Row():
@@ -21,9 +22,9 @@ def process_image(img):
 # Define the Gradio interface
 demo = gr.Interface(
     fn=process_image,
-    inputs=gr.ImageMask(sources=["upload"], layers=False, transforms=[], format="png", label="base image", show_label=True),
+    inputs=gr.ImageMask(sources=["upload"], layers=False, transforms=[], format="png", label="base image", show_label=True, height = 600),
     outputs=[
-        gr.Image(label="Background Image"),
+        # gr.Image(label="Background Image"),
         gr.Image(label="Mask Image"),
     ],
 )