update

app.py

@@ -25,7 +25,7 @@ except Exception as e:
     })
     train_dataset = Dataset.from_dict({'image_id': [], 'caption': []}, features=features)
     val_dataset = Dataset.from_dict({'image_id': [], 'caption': []}, features=features)
-    dataset = DatasetDict({"train": train_dataset, "dev": val_dataset})
+    dataset = DatasetDict({"train": train_dataset, "test": val_dataset})
     dataset.push_to_hub(dataset_name)
 
 image_folder = "images"
@@ -38,7 +38,7 @@ def determine_split(image_id):
     if "train" in image_id:
         return "train"
     elif "val" in image_id:
-        return "dev"
+        return "test"
     else:
         return None
 
@@ -46,7 +46,7 @@ def determine_split(image_id):
 # Function to get a random image that hasn’t been annotated or skipped
 def get_next_image(session_data):
     with lock:
-        annotated_images = set(dataset["train"]["image_id"]) | set(dataset["dev"]["image_id"])
+        annotated_images = set(dataset["train"]["image_id"]) | set(dataset["test"]["image_id"])
         available_images = [img for img in image_files if img not in annotated_images]
 
         if session_data["current_image"] is None and available_images: