updates

app.py

@@ -22,12 +22,15 @@ try:
 
     # Update annotation_count for existing images
     def update_annotation_count(example):
-        example["annotation_count"] = 1 if example["annotation_count"] == 0 else example["annotation_count"]
+        if example["annotation_count"] == 0:
+            example["annotation_count"] = 1
         return example
 
     dataset = dataset.map(update_annotation_count)
     print("Updated annotation counts for existing images.")
+    dataset.push_to_hub(dataset_name)  # Push the updated dataset to Hugging Face
 except Exception as e:
+    print(f"Error loading dataset: {e}")
     # Create an empty dataset if it doesn't exist
     features = Features({
         'image_id': Value(dtype='string'),