x

app.py

@@ -51,15 +51,10 @@ else:
     if idx == index:
         cap = f'{idx}'
 
-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
-    caps[idx] = f'{idx} (TARGET IMAGE)'
-    if idx == index:
-        cap = f'{idx} (TARGET IMAGE)'
 if col1.button('Hide groundtruth target image'):
     st.session_state.show = False
-    caps[idx] = f'{idx}'
-    if idx == index:
-        cap = f'{idx}'
+
+col1.image(img, use_column_width=True, caption=cap)
+col2.image(images, width=175, caption=caps)