Add simple gradio to create image masks

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+
+
+def draw_mask(args):
+    init_image = args.get("image")
+    mask = args.get("mask")
+    mask = mask.convert("RGB").resize((512, 512))
+    init_image = init_image.convert("RGB").resize((512, 512))
+    return mask
+
+
+gr.Interface(
+    fn=draw_mask,
+    inputs=gr.Image(
+        source="upload",
+        tool="sketch",
+        elem_id="image_upload",
+        type="pil",
+        label="Upload",
+    ).style(height=400),
+    outputs=gr.Image(label="Result"),
+).launch()

pyproject.toml

@@ -0,0 +1,19 @@
+[tool.poetry]
+name = "masquerade"
+version = "0.1.0"
+description = ""
+authors = ["David Lundgren <david.m.lundgren@gmail.com>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.11"
+gradio = "^3.22.1"
+torch = "^2.0.0"
+
+
+[tool.poetry.group.dev.dependencies]
+black = "^23.1.0"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"