updates

app.py

@@ -49,8 +49,8 @@ def patch(ij):
 
   imm=ds[int(immg)]['image'].resize(size=(512,512))
 
-  y=p//gridsize
-  x=p%gridsize
+  x=p//gridsize
+  y=p%gridsize
   imc=imm.crop(((x)*stride,(y)*stride,(x+2)*stride,(y+2)*stride))
   return imc
 
@@ -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[1]*gridsize+point[0]
+  p=point[0]*gridsize+point[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=[])