updates

app.py

@@ -27,7 +27,7 @@ gridsize=31
 
 n_patches=961
 
-pred_dict={'Trained on color images': np.load('pred_all_grey.npy').reshape(-1,64),
+pred_dict={'Trained on color images': np.load('pred_all.npy').reshape(-1,64),
            'Trained on greyscale images': np.load('pred_all_grey.npy').reshape(-1,64)}