GeorgeIbrahim commited on
Commit
3d040cd
·
1 Parent(s): 3dde2a6
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -143,8 +143,8 @@ def save_annotation(caption, session_data):
143
  image_id = session_data["current_image"]
144
 
145
  match = re.search(r'_(\d+)\.', image_id)
146
- image_id = match.group(1).lstrip('0')
147
- split = "dev" if image_id in results else "train"
148
 
149
  # Save caption or "skipped" based on user input
150
  if caption.strip().lower() == "skip":
 
143
  image_id = session_data["current_image"]
144
 
145
  match = re.search(r'_(\d+)\.', image_id)
146
+ image_2 = match.group(1).lstrip('0')
147
+ split = "dev" if image_2 in results else "train"
148
 
149
  # Save caption or "skipped" based on user input
150
  if caption.strip().lower() == "skip":