Eun0 commited on
Commit
c117fae
·
1 Parent(s): 9694d0b

First init

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+ def draw_mask(img, h, w):
4
+ return [
5
+ img['image'].convert("RGB").resize((h, w))
6
+ img['mask'].convert("RGB").resize((h, w))
7
+ ]
8
+
9
+ iface = gr.Interface(
10
+ fn=draw_mask,
11
+ inputs=[
12
+ gr.Image(source="upload", tool='sketch', type='pil'),
13
+ gr.Slider(label="height", minimum=256, maximum=1024, value=512),
14
+ gr.Slider(label="width", minimum=256, maximum=1024, value=512),
15
+ ], outputs=[
16
+ gr.Image(),
17
+ gr.Image()
18
+ ])
19
+ iface.launch()