app.py updates

app.py

@@ -42,10 +42,10 @@ if "img" not in st.session_state:
 if "draw" not in st.session_state:
   st.session_state["draw"] = False
 
-def patch(i):
-  st.write(i)
-  img=i//(36*36)
-  p=i%(36*36)
+def patch(ij):
+  st.write(ij)
+  img=ij//(36*36)
+  p=ij%(36*36)
 
   imm=ds[img]['image']
 
@@ -73,8 +73,10 @@ 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)
-  for i in range(4):
-    st.session_state["sideimg"][i]=patch(diff.argsort()[i])
+  for ix in range(4):
+    st.session_state["sideimg"][ix]=patch(diff.argsort()[ix])
+    st.write(diff.argsort()[ix])
+