Spaces:

yuragoithf
/

mlg_semantic_segmentation

Sleeping

yuragoithf commited on Sep 30, 2024

Commit

c48b854

verified ·

1 Parent(s): 964137a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -82,7 +82,7 @@ title = "<h1 style='text-align: center;'>Semantic Segmentation (Airbus Ship Dete
 description = "Upload an image and get prediction mask"
 gr.Interface(fn=gen_pred,
-             inputs=[gr.Image(type='pil')],
              outputs=["plot"],
              title=title,
              examples=[["00c3db267.jpg"], ["00dc34840.jpg"], ["00371aa92.jpg"]],

 description = "Upload an image and get prediction mask"
 gr.Interface(fn=gen_pred,
+             inputs=[gr.components.Image(type='pil')],
              outputs=["plot"],
              title=title,
              examples=[["00c3db267.jpg"], ["00dc34840.jpg"], ["00371aa92.jpg"]],