updates

app.py

@@ -27,6 +27,8 @@ try:
     # Check if the 'split' column exists; if not, add it
     if 'split' not in dataset.column_names:
         # Define the 'split' values based on `image_id`
+        print(results)
+        print(if example["image_id"] in results)
         split_values = [
             "dev" if example["image_id"] in results else "train" 
             for example in dataset