Update app.py

app.py

@@ -2,16 +2,16 @@ import gradio as gr
 import imageio
 
 def dummy(img):
-    imageio.imwrite("output_image.png", img["mask"])
-    return img["mask"]  # 返回基础图像和mask图像
+  imageio.imwrite("output_image.png", img["mask"])
+  return img["image"], img["mask"]
 
 with gr.Blocks() as demo:
-    # 每个图像组件独占一行
-    with gr.Row():
-        img = gr.Image(tool="sketch", label="Base image", show_label=True)
-    with gr.Row():
-        img2 = gr.Image(label="Mask image", show_label=True)
-    btn = gr.Button()
-    btn.click(dummy, img, [img2])  # img1已从这里移除
+  with gr.Row():
+    img = gr.Image(tool="sketch", label="base image", show_label=True)
+  with gr.Row():
+    img1 = gr.Image()
+    img2 = gr.Image(label="mask image", show_label=True)
+  btn = gr.Button()
+  btn.click(dummy, img, [img1, img2])
 
-demo.launch(debug=True)
+demo.launch(debug=True)