Update README.md

README.md

@@ -43,7 +43,7 @@ It serves as a benchmark for performance for self-supervised models.
 
 Use the code below to get started with the model.
 
-```
+```python
 from transformers import AutoModel
 import torch
 import torch.nn as nn
@@ -153,7 +153,7 @@ We have utilized the self-supervised learning framework called DINO. We pre-trai
 
 We used three transforms mainly for preprocessing: SaturationNoiseInjector(), SelfImageNormalize(), Resize(224,224)
 
-```
+```python
 # Noise Injector transformation
 class SaturationNoiseInjector(nn.Module):
     def __init__(self, low=200, high=255):