app.py updates

app.py

@@ -48,7 +48,8 @@ def find():
   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.flatten().argsort()[:4])
+  for i in range(4):
+    st.write(diff.flatten().argsort()[i])
 
  # for i in range(4):
  #   st.session_state["sideimg"][i]+=1