Update README.md

README.md

@@ -15,9 +15,14 @@ Currently pretraining on [JackBAI/bert_pretrain_datasets](https://huggingface.co
 
 
 ```python
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from transformers import AutoModelForMaskedLM, AutoConfig
 
-model_path = 'isemmanuelolowe/BerKANT_171M'
-model = AutoModelForSequenceClassification.from_pretrained(model_path, trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained(model_path)
+# Define the model path
+model_path = "isemmanuelolowe/BerKANT_171M"
+
+# Load the configuration
+config = AutoConfig.from_pretrained(model_path)
+
+# Load the model with the correct configuration
+model = AutoModelForMaskedLM.from_pretrained(model_path, config=config, trust_remote_code=True)
 ```