Update app.py

app.py

@@ -16,7 +16,8 @@ if 'example_idx' not in st.session_state:
     st.session_state.example_idx = 0
 
 st.set_page_config(layout="wide")
-st.markdown("**This is a demo of the *ImageCoDe* dataset. Sample an example on the left and compare all the images with index counter on the right. At the bottom there are buttons to show/hide the groundtruth!**")
+st.markdown("**This is a demo of the *ImageCoDe* benchmark. What is the task? You are given a description and you have to pick the image it describes, out of 10 images total.**")
+st.markdown("**If you click the Sample button, you will get a new text and images. More details of ImageCoDe can be found in our ACL 2022 paper.**")
 
 col1, col2 = st.columns(2)
 
@@ -25,7 +26,7 @@ set2ids = json.load(open('set2ids.json', 'r'))
 descriptions = json.load(open('valid_list.json', 'r'))
 
 #example_idx = int(col1.number_input('Sample an example (description + corresponding images) from the validation set', value=0, min_value=0, max_value=len(descriptions)-1))
-if col1.button('Sample an example (description + corresponding images) from the validation set'):
+if col1.button('Sample a description + 10 images from the validation set'):
     st.session_state.example_idx += 1
 #    st.session_state.example_idx = random.randint(0, len(descriptions)-1)
 
@@ -35,7 +36,7 @@ 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('Click to reveal/hide groundtruth image index (try to guess yourself first!)'):
+if col1.button('Toggle to reveal/hide the correct image, try to guess yourself before giving up!)'):
     st.session_state.show = not st.session_state.show
 
 col1.markdown(f'**Description for {img_set}**:')