dzxcz commited on
Commit
bb0bb40
·
verified ·
1 Parent(s): e124cac

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from PIL import Image, ImageDraw
3
+
4
+ def generate_masked_image(dict):
5
+ mask = dict["mask"].convert("RGB")
6
+ return mask
7
+
8
+ iface = gr.Interface(
9
+ fn=generate_masked_image,
10
+ inputs=[
11
+ gr.inputs.Image(label="Original", type="pil", tool="sketch")
12
+ ],
13
+ outputs=gr.outputs.Image(label="Mask", type="pil")
14
+ )
15
+
16
+ iface.launch()