yuragoithf commited on
Commit
d1eaab8
·
1 Parent(s): 56df16f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ description = "Upload an image and get prediction mask"
77
  # css_code='body{background-image:url("file=wave.mp4");}'
78
 
79
  gr.Interface(fn=gen_pred,
80
- inputs=[gr.Image(type='numpy')],
81
  outputs=gr.Image(type='numpy'),
82
  title=title,
83
  examples=[["003e2c95d.jpg"], ["003b50a15.jpg"], ["003b48a9e.jpg"], ["0038cbe45.jpg"], ["00371aa92.jpg"]],
 
77
  # css_code='body{background-image:url("file=wave.mp4");}'
78
 
79
  gr.Interface(fn=gen_pred,
80
+ inputs=[gr.Image(type='pil')],
81
  outputs=gr.Image(type='numpy'),
82
  title=title,
83
  examples=[["003e2c95d.jpg"], ["003b50a15.jpg"], ["003b48a9e.jpg"], ["0038cbe45.jpg"], ["00371aa92.jpg"]],