Update app.py

app.py

@@ -41,8 +41,8 @@ def inference(input_image):
     # plot the semantic segmentation predictions of 21 classes in each color
     r = Image.fromarray(output_predictions.byte().cpu().numpy()).resize(input_image.size)
     r.putpalette(colors)
-    plt.imshow(r)
-    return plt
+    plt.savefig('out.png')
+    return 'out.png'
 
 
 title = "FCN-RESNET101"
@@ -52,7 +52,7 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1605.062
 gr.Interface(
     inference, 
     gr.inputs.Image(type="pil", label="Input"), 
-    gr.outputs.Image(type="plot", label="Output"),
+    gr.outputs.Image(type="file", label="Output"),
     title=title,
     description=description,
     article=article,