app.py updates

app.py

@@ -11,6 +11,7 @@ ds = load_dataset("Circularmachines/batch_indexing_machine_100_small_imgs", spli
 
 
 pred=np.load('pred.npy')
+pred_all=np.load('pred_all.npy')
 keep_bool=np.load('keep_bool.npy')
 
 
@@ -43,7 +44,7 @@ def find():
   point=(point[0]//16,point[1]//16)
   #point=point[0]*36+point[1]
   st.write(point)
-  st.write(keep_bool[point[0],point[1]])
+  st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])