x

app.py

@@ -55,5 +55,11 @@ 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}'