Update README.md

README.md

@@ -6,9 +6,12 @@ I do not own, distribute, or take credits for this model, all copyrights belong
 
 
 # how to load the model 
-```python
+download the weights
+```
 !git clone https://huggingface.co/not-lain/TunBERT
-!pip install transformers
+```
+load the model
+```python
 import torch.nn as nn
 import torch
 from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModelForSequenceClassification, PreTrainedModel,AutoConfig, BertModel
@@ -44,7 +47,7 @@ class TunBERT(PreTrainedModel):
 
 
 tunbert = TunBERT(config)
-tunbert.load_state_dict(torch.load("/content/TunBERT/pytorch_model.bin"))
+tunbert.load_state_dict(torch.load("TunBERT/pytorch_model.bin"))
 
 tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 ```