Update app.py

app.py

@@ -41,6 +41,7 @@ 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)
     plt.savefig('out.png')
     return 'out.png'