Upload app.py

app.py

@@ -216,9 +216,10 @@ model_inpaint = None
 
 def inpaint(image):
 #    image = resize_image(img, res)
-    color = HWC3(image["image"])
-    alpha = image["mask"][:, :, 1]
-    result = np.concatenate([color, alpha], axis=2)
+#    color = HWC3(image["image"])
+#    alpha = image["mask"][:, :, 0:1]
+#    result = np.concatenate([color, alpha], axis=2)
+    result = image["mask"][:, :, 0:1]
     return [result]
 
 block = gr.Blocks().queue()