Update app.py

app.py

@@ -77,7 +77,7 @@ def predict_2(image):
       
         unpatched_prediction = unpatchify(patched_prediction, (image.shape[0], image.shape[1]))
         labels = LABEL_NAMES
-        res = np.bincount(unpatched_prediction)/unpatched_prediction.shape[0]
+        res = np.bincount(unpatched_prediction.reshape(-1))/unpatched_prediction.reshape(-1).shape[0]
         out = dict(list(zip(labels, res)))
         unpatched_prediction = targets_classes_colors[unpatched_prediction]