x

app.py

@@ -31,10 +31,8 @@ images = [prefix+'/'+img_set+'/'+i for i in set2ids[img_set]]
 img_urls = images.copy()
 index = int(col2.number_input('Image Index from 0 to 9', value=0, min_value=0, max_value=9))
 
-if col1.button('Reveal groundtruth image index (try to guess yourself first!)'):
-    st.session_state.show = True
-if col1.button('Hide groundtruth image index'):
-    st.session_state.show = False
+if col1.button('Click to reveal/hide groundtruth image index (try to guess yourself first!)'):
+    st.session_state.show = not st.session_state.show
 
 col1.markdown(f'**Description**:')
 col1.markdown(f'**{descr}**')