updates

app.py

@@ -79,6 +79,9 @@ def patch(ij):
 if "sideimg" not in st.session_state:
   st.session_state["sideimg"] = [patch(i) for i in range(4)]
 
+if "sideix" not in st.session_state:
+  st.session_state["sideix"] = [i for i in range(4)]
+
 def button_click():
     st.session_state["img"]=np.random.randint(100)
     st.session_state["draw"] = False
@@ -94,6 +97,9 @@ def find():
   diff=np.linalg.norm(pred_all[np.newaxis,i*gridsize**2+p,:]-pred_all,axis=-1)
   for ix in range(4):
     st.session_state["sideimg"][ix]=patch(diff.argsort()[ix])
+    st.session_state["sideix"][ix]=diff.argsort()[ix]
+     [i for i in range(4)]
+
     #st.write(diff.argsort()[ix])
 
 
@@ -154,5 +160,6 @@ with col2:
 
   for i in range(4):
     st.image(st.session_state["sideimg"][i].resize((128,128)))
+    st.write(st.session_state["sideix"][i]//n_patches//20)