updates

app.py

@@ -27,9 +27,9 @@ gridsize=31
 
 n_patches=961
 
-#pred=np.load('pred.npy')
-pred_dict=[np.load('pred_all_hue.npy').reshape(-1,64),np.load('pred_all_grey.npy').reshape(-1,64)]
-#pred_all=np.load('pred_all.npy').reshape(-1,64)
+pred_dict={'Trained on color images': np.load('pred_all_grey.npy').reshape(-1,64),
+           'Trained on greyscale images': np.load('pred_all_grey.npy').reshape(-1,64)}
+
 
 random_i=np.load('random.npy')