app.py updates

app.py

@@ -11,7 +11,7 @@ ds = load_dataset("Circularmachines/batch_indexing_machine_100_small_imgs", spli
 
 
 pred=np.load('pred.npy')
-
+keep_bool=np.load('keep_bool.npy')
 
 
 #st.set_page_config(
@@ -40,7 +40,9 @@ def button_click():
 
 def find():
   point=st.session_state["point"]
-  st.write(point[0]//32,point[1]//32)
+  point=(point[0]//32,point[1]//32)
+  point=point[0]*36+point[1]
+  st.write(keep_bool[point])