x

app.py

@@ -42,6 +42,11 @@ images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index]
 caps = list(range(10))
 cap = str(index)
 
+if col1.button('Show groundtruth target image'):
+    st.session_state.show = True
+if col1.button('Hide groundtruth target image'):
+    st.session_state.show = False
+
 if st.session_state.show:
     caps[idx] = f'{idx} (TARGET IMAGE)'
     if idx == index:
@@ -51,10 +56,5 @@ else:
     if idx == index:
         cap = f'{idx}'
 
-if col1.button('Show groundtruth target image'):
-    st.session_state.show = True
-if col1.button('Hide groundtruth target image'):
-    st.session_state.show = False
-
 col1.image(img, use_column_width=True, caption=cap)
 col2.image(images, width=175, caption=caps)