I am just following the notebook which is this:- Transformers-Tutorials/Custom_Named_Entity_Recognition_with_BERT.ipynb at master · NielsRogge/Transformers-Tutorials · GitHub 2
and I only done some small changes which I listed above!