Update README.md

README.md

@@ -21,6 +21,8 @@ BERTikal [1] is a cased BERT-base model for the Brazilian legal language and was
 
 ## Usage
 
+### Ex. Loading model for general use
+
 ```python
 from transformers import AutoTokenizer  # Or BertTokenizer
 from transformers import AutoModelForPreTraining  # Or BertForPreTraining for loading pretraining heads
@@ -30,7 +32,6 @@ model = AutoModelForPreTraining.from_pretrained('felipemaiapolo/legalnlp-bert')
 tokenizer = AutoTokenizer.from_pretrained('felipemaiapolo/legalnlp-bert', do_lower_case=False)
 ```
 
-
 ### Ex. BERT embeddings
 
 ```python
@@ -40,12 +41,12 @@ pipe = pipeline("feature-extraction", model='felipemaiapolo/legalnlp-bert')
 encoded_sentence = pipe('Juíz negou o recurso.')
 ```
 
-### Masked language modeling prediction example
+### Ex. Masked language modeling prediction 
 
 ```python
 from transformers import pipeline
 
-pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer)
+pipe = pipeline('fill-mask', model='felipemaiapolo/legalnlp-bert')
 
 pipe('Juíz negou o [MASK].')
 #  [{'score': 0.6387444734573364,