Spaces:

yuragoithf
/

mlg_semantic_segmentation

Sleeping

yuragoithf commited on Jun 27, 2023

Commit

e54b3e5

1 Parent(s): 5f088d2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -80,7 +80,7 @@ description = "Upload an image and get prediction mask"
 gr.Interface(fn=gen_pred,
              inputs=[gr.Image(type='numpy')],
-             outputs=gr.Image(type='numpy'),
              title=title,
              examples=[["003e2c95d.jpg"], ["003b50a15.jpg"], ["003b48a9e.jpg"], ["0038cbe45.jpg"], ["00371aa92.jpg"]],
              description=description,

 gr.Interface(fn=gen_pred,
              inputs=[gr.Image(type='numpy')],
+             outputs=gr.Image(type='pil'),
              title=title,
              examples=[["003e2c95d.jpg"], ["003b50a15.jpg"], ["003b48a9e.jpg"], ["0038cbe45.jpg"], ["00371aa92.jpg"]],
              description=description,