Update app.py

app.py

@@ -33,6 +33,6 @@ def predict_segmentation(img):
     return output_image
 
 input_image = gr.inputs.Image()
-output_image = gr.outputs.Image()
+output_image = gr.outputs.Image(type='pil')
 app = gr.Interface(fn=predict_segmentation, inputs=input_image, outputs=output_image, title='Microstructure Segmentation', description='Segment the input image into grain and background.')
 app.launch()