The usual workflow would be to add the transformations in the Dataset.__getitem__.
If you need to see the original text (without the transformations), you could return the original as well as the transformed samples from __getitem__.