dc dc

app.py

@@ -119,6 +119,7 @@ def main():
             count = user_state if isinstance(user_state, int) else user_state.value
             if count < config['dataset'][config['dataset']['name']]['n_classes']:
                 img_id = img_ids[-1] if isinstance(img_ids[-1], int) else img_ids[-1].value
+                print(f"Updating saliency maps for class index: {count} and image ID: {img_id}")
                 images = load_image_and_saliency(count, data_dir, img_id)
                 target_img = gr.Image(images[0], elem_classes="main-image", visible=True)
                 saliency_gradcam = gr.Image(images[1], elem_classes="main-image", visible=True)