updates

app.py

@@ -127,14 +127,14 @@ def save_annotation(caption, session_data):
         return gr.update(visible=False), gr.update(value="All images have been annotated!"), gr.update(value="")
 
     with lock:
-        match = re.search(r'_(\d+)\.', image_path)
         image_id = session_data["current_image"]
         print("Image ID before: ", image_id)
-        
+                
+        match = re.search(r'_(\d+)\.', image_id)
         image_id_2 = match.group(1).lstrip('0') 
         split = "dev" if image_id_2 in results else "train"
         print("Image ID after: ", image_id_2)
-        
+
         # Save caption or "skipped" based on user input
         if caption.strip().lower() == "skip":
             caption = "skipped"