Update app.py

app.py

@@ -35,8 +35,8 @@ def predict_segmentation(image: Image.Image):
 
     return output_image
 
-image_input = gr.inputs.Image(shape=(256, 256), source="upload")
-image_output = gr.outputs.Image(type="pil")
+image_input = gr.Image(shape=(256, 256), source="upload")
+image_output = gr.Image(type="pil")
 
 iface = gr.Interface(fn=predict_segmentation, inputs=image_input, outputs=image_output)
 iface.launch()