Update app.py

app.py

@@ -165,6 +165,8 @@ if __name__ == '__main__':
             else:
                 pred_mask = np.argmax(preds, axis=-1)
 
+            pred_mask = decode_from_colormap(pred_mask, dataset.colors)[..., ::-1]
+
             return pred_mask.astype(np.uint8)