Inside the Token classification pipeline (PyTorch)

Yes, I got it. Thanks! However, the final output of this piece of code doesn’t perform as expected. Here is the link to the code.