yuragoithf commited on
Commit
8d2b059
·
1 Parent(s): 4b7b314

Update app.py

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