Update README.md

README.md

@@ -27,7 +27,7 @@ This repository contains a BERT-based model, **medBERT-base**, fine-tuned on the
 
 You can load the pre-trained **medBERT-base** model using the Hugging Face `transformers` library:
 
-'''
+```py
 from transformers import BertTokenizer, BertForMaskedLM
 import torch
 
@@ -49,7 +49,7 @@ top_k_tokens = tokenizer.convert_ids_to_tokens(top_k_ids)
 print("Top 5 prediction:")
 for i, token in enumerate(top_k_tokens):
     print(f"{i + 1}: {token}")
-'''
+```
 
 _Top 5 prediction:_
 _1: from_