hysts HF Staff commited on
Commit
36711e2
·
1 Parent(s): a239782
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,8 +60,8 @@ demo = gr.Interface(
60
  fn=predict,
61
  inputs=gr.Image(label="Input", type="pil"),
62
  outputs=[
63
- gr.Image(label="Predicted Labels", type="numpy"),
64
- gr.Image(label="Masked", type="numpy"),
65
  ],
66
  examples=examples,
67
  title=TITLE,
 
60
  fn=predict,
61
  inputs=gr.Image(label="Input", type="pil"),
62
  outputs=[
63
+ gr.Image(label="Predicted Labels"),
64
+ gr.Image(label="Masked"),
65
  ],
66
  examples=examples,
67
  title=TITLE,