Create app.py

app.py

@@ -38,8 +38,9 @@ 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
+    return r
+    # plt.imshow(r)
+    # return plt
 
 title = "DEEPLABV3-RESNET101"
 description = "demo for DEEPLABV3-RESNET101, DeepLabV3 model with a ResNet-101 backbone. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
@@ -48,7 +49,8 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1706.055
 gr.Interface(
     inference, 
     gr.inputs.Image(type="pil", label="Input"), 
-    gr.outputs.Image(type="plot", label="Output"),
+    gr.outputs.Image(type="pil", label="Output"),
+    # gr.outputs.Image(type="plot", label="Output"),
     title=title,
     description=description,
     article=article,