app.py updates

app.py

@@ -49,8 +49,8 @@ def patch(ij):
 
   imm=ds[int(immg)]['image']
 
-  y=p//36
-  x=p%36
+  x=p//36
+  y=p%36
   imc=imm.crop((x*16,y*16,(x+1)*16,(y+1)*16))
   return imc
 
@@ -72,7 +72,7 @@ def find():
   #st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
   i=st.session_state["img"]
   p=point[0]*36+point[1]
-  diff=np.linalg.norm(pred_all[np.newaxis,np.newaxis,i,p]-pred_all,axis=-1).flatten()
+  diff=np.linalg.norm(pred_all[np.newaxis,np.newaxis,i,p,:]-pred_all,axis=-1).flatten()
   for ix in range(4):
     st.session_state["sideimg"][ix]=patch(diff.argsort()[ix])
     #st.write(diff.argsort()[ix])