AutoTrain Token Classification Error

🤗AutoTrain
1

My sample data .csv:

image
image769Ă—43 1.93 KB

Column mapping in AutoTrain:
{“text”: “tokens”, “label”: “tags”}

❌ ERROR  | 2024-03-12 06:53:38 | autotrain.trainers.common:wrapper:91 - train has failed due to an exception: Traceback (most recent call last):
  File "/app/env/lib/python3.10/site-packages/autotrain/trainers/common.py", line 88, in wrapper
    return func(*args, **kwargs)
  File "/app/env/lib/python3.10/site-packages/autotrain/trainers/token_classification/__main__.py", line 68, in train
    label_list = train_data.features[config.tags_column].feature.names
KeyError: 'tags'

❌ ERROR  | 2024-03-12 06:53:38 | autotrain.trainers.common:wrapper:92 - 'tags'