updates

app.py

@@ -40,9 +40,9 @@ except Exception as e:
         'image_id': Value(dtype='string'),
         'caption': Value(dtype='string'),
         'annotation_count': Value(dtype='int32'),
-        'split': Value(dtype='string')
+        'spl': Value(dtype='string')
     })
-    dataset = Dataset.from_dict({'image_id': [], 'caption': [], 'annotation_count': [], 'split': []}, features=features)
+    dataset = Dataset.from_dict({'image_id': [], 'caption': [], 'annotation_count': [], 'spl': []}, features=features)
     annotation_counts = {}
     dataset.push_to_hub(dataset_name)  # Push the empty dataset to Hugging Face
 
@@ -122,12 +122,12 @@ def save_annotation(caption, session_data):
             "image_id": [image_id],
             "caption": [caption],
             "annotation_count": [annotation_count + 1],
-            "split": [split]
+            "spl": [split]
         }, features=Features({
             'image_id': Value(dtype='string'),
             'caption': Value(dtype='string'),
             'annotation_count': Value(dtype='int32'),
-            'split': Value(dtype='string')
+            'spl': Value(dtype='string')
         }))
 
         # Update the annotation count in the dictionary