Spaces:
Sleeping
Sleeping
Commit
·
3d040cd
1
Parent(s):
3dde2a6
updates
Browse files
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 |
-
|
| 147 |
-
split = "dev" if
|
| 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":
|