Update README.md

README.md

@@ -100,6 +100,8 @@ def predict_mask(sent, tokenizer, model, topk=10, print_results=True):
             print(f"Mask Predictions: {predictions}")
 
     return words
+```
+
 
 # Reference
 @inproceedings{aghaei2023securebert,