updates

app.py

@@ -27,7 +27,7 @@ gridsize=31
 
 n_patches=961
 
-pred=np.load('pred.npy')
+#pred=np.load('pred.npy')
 pred_all=np.load('pred_all.npy').reshape(-1,64)
 
 random_i=np.load('random.npy')
@@ -62,7 +62,7 @@ def find():
   #st.write(point)
   #st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
   i=st.session_state["img"]
-  p=point[0]*gridsize+point[1]
+  p=point[1]*gridsize+point[0]
   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=[])