You can decide to test either from a set of messages that you have prepared manually, or you can use the training, or you can use the validation split of a test set that's on Hugging Face by setting use data set to test equal to true right here. Next, you set up your training and validation splits. Here I've selected a main branch for training, and I've selected the training split. You can fix a max number of rows here. This will save you time if you just want to download and run on, say, 100 rows instead of on a massive dataset.