updates

app.py

@@ -55,6 +55,21 @@ try:
 
     print("Annotation counts:", annotation_counts)
 
+
+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'),
+        'caption': Value(dtype='string'),
+        'annotation_count': Value(dtype='int32'),
+        'split': Value(dtype='string')
+    })
+    dataset = Dataset.from_dict({'image_id': [], 'caption': [], 'annotation_count': [], 'split': []}, features=features)
+    annotation_counts = {}
+    dataset.push_to_hub(dataset_name)  # Push the empty dataset to Hugging Face
+
+
 # Initialize or reset data as needed based on the `reset` flag
 if reset:
     # Clear the annotation counts
@@ -79,19 +94,6 @@ if reset:
     dataset.push_to_hub(dataset_name)
     print("Data has been reset.")
 
-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'),
-        'caption': Value(dtype='string'),
-        'annotation_count': Value(dtype='int32'),
-        'split': Value(dtype='string')
-    })
-    dataset = Dataset.from_dict({'image_id': [], 'caption': [], 'annotation_count': [], 'split': []}, features=features)
-    annotation_counts = {}
-    dataset.push_to_hub(dataset_name)  # Push the empty dataset to Hugging Face
-
 image_folder = "images"
 image_files = [f for f in os.listdir(image_folder) if f.endswith(('.png', '.jpg', '.jpeg'))]
 lock = threading.Lock()