Update README.md

README.md

@@ -59,7 +59,7 @@ for p in preds:
 
 ### 2. Use as an Encoder (CLS Embeddings)
 
-```
+```python
 import torch
 from transformers import AutoTokenizer, AutoModel
 
@@ -91,7 +91,7 @@ print(embedding.shape)  # torch.Size([768])
 
 ### 3. Compute Similarity Between Two Articles
 
-```
+```python
 import torch.nn.functional as F
 
 e1 = encode("Article text one...")