updates

app.py

@@ -13,7 +13,7 @@ st.set_page_config(layout="wide")
 
 
 
-ds = load_dataset("Circularmachines/batch_indexing_machine_test", split="test")
+ds = load_dataset("Circularmachines/batch_indexing_machine_green_test", split="test")
 
 
 
@@ -27,8 +27,8 @@ gridsize=31
 
 n_patches=961
 
-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)}
+pred_dict={'Trained on color images': np.load('pred_all_green.npy').reshape(-1,64),
+           'Trained on greyscale images': np.load('pred_all_green.npy').reshape(-1,64)}
 
 
 random_i=np.load('random.npy')

pred_all_green.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6287951aee65bb32b679a1e5173e95407158b277a7db7cedea186867c3ea417
+size 206653568