Spaces:

akhaliq
/

U-2-Net

Runtime error

Ahsen Khaliq commited on Oct 21, 2021

Commit

ac14ec4

1 Parent(s): 2597891

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def infer(img):
       visualization=True)
   return result[0]['front'][:,:,::-1], result[0]['mask']
-inputs = gr.inputs.Image(source="webcam", label="Webcam")
 outputs = [
            gr.outputs.Image(type="numpy",label="Front"),
            gr.outputs.Image(type="numpy",label="Mask")

       visualization=True)
   return result[0]['front'][:,:,::-1], result[0]['mask']
+inputs = gr.inputs.Image(source="webcam", label="Webcam", type="pil")
 outputs = [
            gr.outputs.Image(type="numpy",label="Front"),
            gr.outputs.Image(type="numpy",label="Mask")