How to prepare the training dataset for a trainer of BERT in pytorch?

Question

The task is to detect whether a blood label is present in a text sequence using BERT for sequence classification pre-trained model.

class BloodDataset(Dataset