Update README.md

README.md

@@ -51,3 +51,14 @@ The following hyperparameters were used during training:
 - Pytorch 2.2.1+cu121
 - Datasets 2.18.0
 - Tokenizers 0.15.2
+
+
+```
+from transformers import BertForSequenceClassification, TrainingArguments, Trainer, AutoTokenizer, DataCollatorWithPadding
+model_id = "bert-base-uncased"
+model = BertForSequenceClassification.from_pretrained(model_id, num_labels=3)
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+def tokenize(batch):
+    return tokenizer(batch["text"], truncation=True, padding="max_length", max_length=max_source_length, add_special_tokens=True, return_tensors='pt')
+```