updates

app.py

@@ -111,7 +111,11 @@ def save_annotation(caption, session_data):
             "image_id": [image_id],
             "caption": [caption],
             "annotation_count": [annotation_count + 1]  # Increment the annotation count
-        })
+        }, features=Features({
+            'image_id': Value(dtype='string'),
+            'caption': Value(dtype='string'),
+            'annotation_count': Value(dtype='int32')  # Ensure int32 type
+        }))
 
         # Concatenate with the existing dataset and push the updated dataset to Hugging Face
         dataset = concatenate_datasets([dataset, new_data])