Update app.py

app.py

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