x

app.py

@@ -22,11 +22,6 @@ 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'))
 
-if col1.button('Show groundtruth target image'):
-    st.session_state.show = True
-if col2.button('Hide groundtruth target image'):
-    st.session_state.show = False
-
 # 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'):
     st.session_state.example_idx = random.randint(0, len(descriptions)-1)
@@ -43,6 +38,7 @@ col1.markdown(descr)
 img = images[index]
 images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=20,fill='blue')
 
+
 caps = list(range(10))
 cap = str(index)
 
@@ -57,3 +53,7 @@ else:
 
 col1.image(img, use_column_width=True, caption=cap)
 col2.image(images, width=175, caption=caps)
+if col1.button('Show groundtruth target image'):
+    st.session_state.show = True
+if col1.button('Hide groundtruth target image'):
+    st.session_state.show = False