update app.py

app.py

@@ -22,8 +22,8 @@ def segment_sidewalk(image):
     prediction = (prob_map > 0.5).float()
     prob_map, prediction = prob_map.numpy(), prediction.numpy()
     save_image(image, 'image.png')
-    save_image(prob_map, 'prob.png', cmap='jet')
     save_image(prediction, 'mask.png', cmap='gray')
+    save_image(prob_map, 'prob.png', cmap='jet')
     return Image.open('image.png'), Image.open('mask.png'), Image.open('prob.png')
 
 def save_image(image, path, **kwargs):