add app.py

app.py

@@ -0,0 +1,26 @@
+import os
+import gradio as gr
+from PIL import Image, ImageOps
+
+root = os.path.dirname(os.path.abspath(__file__))
+static = os.path.join(root, "static")
+
+def save_mask(inputs):
+    layers = inputs['layers']
+    if not layers:
+        return inputs['background']
+    mask = layers[0]
+    new_image = Image.new('RGBA', mask.size, color='white')
+    new_image.paste(mask, mask=mask)
+    new_image = new_image.convert('RGB')
+    return ImageOps.invert(new_image)
+
+
+ui = gr.Interface(
+    save_mask,
+    gr.ImageMask(type='pil'),
+    "image"
+)
+
+if __name__ == "__main__":
+    ui.launch(share=False)