app.py updates

app.py

@@ -43,10 +43,12 @@ def find():
   point=st.session_state["point"]
   point=(point[0]//16,point[1]//16)
   #point=point[0]*36+point[1]
-  st.write(point)
-  st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
-
-
+  #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]
+  diff=np.linalg.norm(pred_all[np.newaxis,np.newaxis,i,p]-pred_all,axis=-1)
+  st.write(diff.argmin())
 
  # for i in range(4):
  #   st.session_state["sideimg"][i]+=1