Spaces:

biennh
/

interior-segmentation

Runtime error

biennh commited on Nov 4, 2023

Commit

4261ef7

1 Parent(s): f3beccb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def segment(input_img):
 demo = gr.Interface(fn=segment,
-                    inputs=gr.Image(shape=(400, 600)),
                     outputs=['image'],
                     allow_flagging='never')
 if __name__ == "__main__":

 demo = gr.Interface(fn=segment,
+                    inputs=gr.Image(type='pil'),
                     outputs=['image'],
                     allow_flagging='never')
 if __name__ == "__main__":