app.py updates

app.py

@@ -54,8 +54,6 @@ def patch(ij):
   imc=imm.crop((x*16,y*16,(x+1)*16,(y+1)*16))
   return imc
 
-
-
 if "sideimg" not in st.session_state:
   st.session_state["sideimg"] = [patch(i) for i in range(4)]
 
@@ -63,7 +61,6 @@ def button_click():
     st.session_state["img"]=np.random.randint(100)
     st.session_state["draw"] = False
 
-
 def find():
   point=st.session_state["point"]
   point=(point[0]//16,point[1]//16)
@@ -71,7 +68,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]*36+point[1]
+  p=point[1]*36+point[0]
   diff=np.linalg.norm(pred_all[np.newaxis,i*36*36+p,:]-pred_all,axis=-1)
   for ix in range(4):
     st.session_state["sideimg"][ix]=patch(diff.argsort()[ix])