resize image

app.py

@@ -10,6 +10,6 @@ def segment(input_img):
     return mask_image
 
 
-demo = gr.Interface(segment, gr.Image(), "image")
+demo = gr.Interface(segment, gr.Image(shape=(400, 600)), "image")
 if __name__ == "__main__":
     demo.launch()