Update README.md

README.md

@@ -72,11 +72,11 @@ The model demonstrated continuous improvement in reducing both training and vali
 ### Usage
 You can load the model from the Hugging Face Hub using the following code:
 
-```
+```python
 # Ensure you have the necessary dependencies installed:
 pip install torch torchvision transformers
 
 from transformers import AutoModel
 
 model = AutoModel.from_pretrained("sebastiansarasti/AutoEncoderImageColorization")
-```python
+```