Spaces:

yuragoithf
/

mlg_semantic_segmentation

Sleeping

yuragoithf commited on Jun 28, 2023

Commit

40dced2

1 Parent(s): 653cbef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ description = "Upload an image and get prediction mask"
 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,

 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,