dimension fix

app.py

@@ -61,7 +61,7 @@ def predict(ash_image, model=model):
     #ash_image = np.load(ash_image)[...,4]
     ash_image = np.asarray(ash_image)
     y_pred = model.predict(ash_image.reshape(1,256, 256, 3))
-    prediction = np.argmax(y_pred[0], axis=2).reshape(256,256,1)
+    prediction = np.argmax(y_pred[0], axis=2).reshape(256,256)
     #intersection = label & prediction
     #false_negative = label - intersection
     #false_possitive = prediction - intersection