Update README.md

README.md

@@ -26,8 +26,13 @@ in medical radiology reports.
 ```python
 from transformers import AutoTokenizer, AutoModelForTokenClassification
 
+ner_labels = ['B-ABNORMALITY', 'I-ABNORMALITY', 'B-NON-ABNORMALITY', 'I-NON-ABNORMALITY', 'B-DISEASE', 'I-DISEASE', 'B-NON-DISEASE', 'I-NON-DISEASE', 'B-ANATOMY', 'I-ANATOMY', 'O']
+
 tokenizer = AutoTokenizer.from_pretrained("Angelakeke/RaTE-NER-Deberta")
-model = AutoModelForTokenClassification.from_pretrained("Angelakeke/RaTE-NER-Deberta")
+model = AutoModelForTokenClassification.from_pretrained("Angelakeke/RaTE-NER-Deberta",
+                                                        num_labels=len(ner_labels),
+                                                        ignore_mismatched_sizes=True,
+                                                        )
 ```
 
 ## Author