Error in fine-tuning bert with custom text

Hi, I am using a hugging face transformer to fine-tune the BERT model using my custom text corpus.

But it gives me an error in the training phase -
IndexError: Invalid key: 2664 is out of bounds for size 0

Kindly find the link to google colab here - Fine-tune Bert model

Any help is appreciated. Kindly help me to make through it.