updates

app.py

@@ -19,6 +19,14 @@ dataset_name = "GeorgeIbrahim/EGYCOCO"  # Replace with your dataset name
 try:
     dataset = load_dataset(dataset_name, split="train")
     print("Loaded existing dataset:", dataset)
+
+    # Update annotation_count for existing images
+    def update_annotation_count(example):
+        example["annotation_count"] = 1 if example["annotation_count"] == 0 else example["annotation_count"]
+        return example
+
+    dataset = dataset.map(update_annotation_count)
+    print("Updated annotation counts for existing images.")
 except Exception as e:
     # Create an empty dataset if it doesn't exist
     features = Features({
@@ -73,7 +81,7 @@ def get_next_image(session_data):
         available_images = [
             img for img in image_files
             if img not in annotated_images or
-               ("val" in img and annotated_images[img] < 2) or
+               ("val" in img and annotated_images[img] < 5) or
                ("val" not in img and annotated_images[img] == 0)
         ]