Eun0 commited on
Commit
acc7af8
·
1 Parent(s): c117fae
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ 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
 
@@ -14,6 +14,6 @@ iface = gr.Interface(
14
  gr.Slider(label="width", minimum=256, maximum=1024, value=512),
15
  ], outputs=[
16
  gr.Image(),
17
- gr.Image()
18
  ])
19
  iface.launch()
 
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
 
 
14
  gr.Slider(label="width", minimum=256, maximum=1024, value=512),
15
  ], outputs=[
16
  gr.Image(),
17
+ gr.Image(),
18
  ])
19
  iface.launch()