app.py updates

app.py

@@ -72,10 +72,10 @@ 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)
+  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])
+    #st.write(diff.argsort()[ix])