Update src/main.py

src/main.py

@@ -95,8 +95,8 @@ class AutoTrain:
 				"dataset_split": "train",
 				"split": 4,
 				"col_mapping": {
-					"image": "image",
-					"label": "target",
+					"text": "contents",
+					"label": "key issue",
 				}
 			},
 			headers={