Update README.md

README.md

@@ -38,9 +38,14 @@ VetBERT was further finetuned on a set of 5002 annotated clinical notes to class
 Load the model via the transformers library:
 
 ```
-from transformers import AutoTokenizer, AutoModel
-tokenizer = AutoTokenizer.from_pretrained("havocy28/VetBERTDx")
-model = AutoModel.from_pretrained("havocy28/VetBERTDx")
+from transformers import AutoTokenizer, AutoModelForMaskedLM
+
+tokenizer = AutoTokenizer.from_pretrained("havocy28/VetBERT")
+model = AutoModelForMaskedLM.from_pretrained("havocy28/VetBERT")
+
+VetBERT_masked = pipeline("fill-mask", model=model, tokenizer=tokenizer)
+VetBERT('Suspected pneuomina, will require an [MASK] but in the meantime will prescribed antibiotics')
+
 ```
 
 ## Citation