x

app.py

@@ -7,6 +7,9 @@ import requests
 import random
 import io
 
+def set_back():
+    index = 0
+
 st.set_page_config(layout="wide")
 st.markdown("**This is a demo of the *ImageCoDe* dataset. Sample an example description with the left +/- button on the right and compare all the images with the +/- button on the right. If you want to know the groundtruth solution, scroll down to the end of the page!**")
 
@@ -25,7 +28,7 @@ descriptions = json.load(open('valid_list.json', 'r'))
 
 
 example = int()
-example_idx = int(col1.number_input('Sample an example from the validation set', value=0, min_value=0, max_value=len(descriptions)-1))
+example_idx = int(col1.number_input('Sample an example from the validation set', value=0, min_value=0, max_value=len(descriptions)-1, on_change=set_back))
 img_set, idx, descr = descriptions[example_idx]
 idx = int(idx)
 images = [prefix+'/'+img_set+'/'+i for i in set2ids[img_set]]