updates

app.py

@@ -64,6 +64,8 @@ def save_annotation(caption, session_data):
             target_dataset = val_dataset
             split_name = "validation"
         else:
+            print(image_id)
+            print("val" in image_id)
             return gr.update(visible=False), gr.update(value="Unknown dataset split for image!")
 
         # Add the new annotation as a new row to the appropriate dataset