Update README.md

README.md

@@ -32,10 +32,8 @@ 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",
-                                                        num_labels=len(ner_labels),
-                                                        ignore_mismatched_sizes=True,
-                                                        )
+model = AutoModelForTokenClassification.from_pretrained("Angelakeke/RaTE-NER-Deberta")
+
 ```
 
 ## Author