x

app.py

@@ -21,8 +21,7 @@ prefix = 'https://raw.githubusercontent.com/BennoKrojer/imagecode-val-set/main/i
 set2ids = json.load(open('set2ids.json', 'r'))
 descriptions = json.load(open('valid_list.json', 'r'))
 
-show = col1.button("Show groundtruth image index")
-hide = col2.button('Hide groundtruth image index')
+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!")
 
 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))
@@ -32,7 +31,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=len(images)-1))
 
-col1.subheader(descr)
+col1.text(f'Description: {descr}')
 
 img = images[index]
 images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=20,fill='blue')
@@ -40,17 +39,7 @@ images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index]
 caps = list(range(10))
 cap = str(index)
 
-if show:
-    if index == idx:
-        cap = f'{index} (GROUNDTRUTH)'
-    caps[idx] = f'{idx} (GROUNDTRUTH)'
-    images[idx] = ImageOps.expand(Image.open(io.BytesIO(requests.get(img_urls[index], stream=True).content)),border=50,fill='green')
-if hide:
-    if index == idx:
-        cap = f'{index}'
-    caps[idx] = f'{idx}'
-    images[idx] = Image.open(io.BytesIO(requests.get(img_urls[index], stream=True).content))
-# shown = True if len(caps[idx]) > 2 else False
-# if shown 
 col1.image(img, use_column_width=True, caption=cap)
 col2.image(images, width=175, caption=caps)
+
+st.markdown('#')