You can check out this thread I just wrote on how to convert predictions to actual labels for token classification models.