Abijith commited on
Commit
e519862
·
1 Parent(s): fe5d1bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -149,10 +149,8 @@ def inference_scribble(image):
149
  return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'),
150
  Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
151
 
152
- demo = gr.Interface(
153
- fn=inference_scribble,
154
- inputs=
155
- gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
156
  outputs=[
157
  gr.outputs.Image(type="pil", label="Mask with Image"),
158
  gr.outputs.Image(type="pil", label="Mask")
 
149
  return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'),
150
  Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
151
 
152
+ demo = gr.Interface(fn=inference_scribble,
153
+ inputs=gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
 
 
154
  outputs=[
155
  gr.outputs.Image(type="pil", label="Mask with Image"),
156
  gr.outputs.Image(type="pil", label="Mask")