updates

app.py

@@ -63,7 +63,7 @@ def find():
   #st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
   i=st.session_state["img"]
   p=point[1]*gridsize+point[0]
-  diff=np.linalg.norm(pred_all[np.newaxis,i*gridsize**2+p,:]-pred_all,axis=-1)
+  diff=np.linalg.norm(pred_all[np.newaxis,i*n_patches+p,:]-pred_all,axis=-1)
   #re_pred=pred_all.reshape(20,20,256,64)
   #diff_re=diff.reshape((20,20,256)).argmin(axis=[])
   i=0