updates

app.py

@@ -27,9 +27,10 @@ gridsize=31
 
 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 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 from scratch': np.load('pred_all_scratch.npy').reshape(-1,64),
+           'Trained from imagenet': np.load('pred_all_green.npy').reshape(-1,64)}
 
 random_i=np.load('random.npy')