updates

app.py

@@ -184,7 +184,7 @@ with col1:
 with col2:
  # st.write("current selection:")
   for i in range(4):
-    st.image(st.session_state["sideimg"][i].resize((128,128)))
+    
     if i==0:
       st.write("current selection in batch "+str(st.session_state["sideix"][i]))#//(gridsize**2)//20))
     if i==1:
@@ -194,6 +194,7 @@ with col2:
     if i==3:
       st.write("Third best match found in batch "+str(st.session_state["sideix"][i]))#//(gridsize**2)//20))
 
+    st.image(st.session_state["sideimg"][i].resize((128,128)))
     
     #st.write(st.session_state["sideix"][i])