Update app.py

app.py

@@ -61,7 +61,12 @@ def do_prediction(img):
             img_patch = img[index_y_d:index_y_u, index_x_d:index_x_u, :]
             label_p_pred = model.predict(img_patch.reshape(1, img_patch.shape[0], img_patch.shape[1], img_patch.shape[2]),
                                          verbose=0)
+
+            print(label_p_pred.shape,'label_p_pred')
+            print(img_patch.shape, 'img_patch')
             seg = np.argmax(label_p_pred, axis=2)
+
+            print(seg.shape, 'seg')
             seg_color = np.repeat(seg[:, :, np.newaxis], 3, axis=2)
 
             if i == 0 and j == 0: