Update README.md

README.md

@@ -8,7 +8,8 @@ pipeline_tag: feature-extraction
 ---
 
 # Usage
-`from transformers import AutoTokenizer
+```python
+from transformers import AutoTokenizer
 from model import (
     BERTContrastiveLearning_simcse,
     BERTContrastiveLearning_simcse_w,
@@ -29,4 +30,5 @@ model2 = BERTContrastiveLearning_simcse_w.load_from_checkpoint(ckpt2).eval()
 model3 = BERTContrastiveLearning_samp.load_from_checkpoint(ckpt3).eval()
 model4 = BERTContrastiveLearning_samp_w.load_from_checkpoint(ckpt4).eval()
 
-cls, _ = model(input_ids, attention_mask)  # embeddings`
+cls, _ = model(input_ids, attention_mask)  # embeddings
+```