updates

app.py

@@ -43,6 +43,7 @@ n_patches=len(keep)
 pred=np.load('pred.npy')
 pred_all=np.load('pred_all.npy').reshape(-1,64)
 
+random_i=np.load('random.npy')
 
 
 #st.set_page_config(
@@ -179,8 +180,14 @@ with col1:
   #subcol1, subcol2 = st.columns(2)
   #with subcol1:
   #st.button('Previous Frame', on_click=button_click)
-  st.button('Change Image', on_click=button_click)
-  st.button('Find similar parts', on_click=find)
+  scol1, scol2 = st.columns(2)
+  with scol1:
+    st.button('Change Image', on_click=button_click)
+
+  with scol2: 
+    st.button('Find similar parts', on_click=find)
+
+  st.write("Currently viewing frame "+random_i[st.session_state["img"]%20]+" in batch "+st.session_state["img"]//20)
 
   #st.write(st.session_state["img"])
   #st.write(st.session_state["point"])