Update app.py

app.py

@@ -29,8 +29,8 @@ n_patches=961
 
 #pred_dict={'Trained on color images (recommended)': np.load('pred_all_scratch.npy').reshape(-1,64),
 #           'Trained on grayscale images': np.load('pred_all_grey.npy').reshape(-1,64)}
-pred_dict={'Trained on normal': np.load('pred_all_scratch.npy').reshape(-1,64),
-           'Trained on augmented': np.load('pred_all_green_random.npy').reshape(-1,64)}
+pred_dict={'Trained on unaugmented images': np.load('pred_all_scratch.npy').reshape(-1,64),
+           'Trained on augmented images': np.load('pred_all_green_random.npy').reshape(-1,64)}
 
 random_i=np.load('random.npy')