Update README.md

README.md

@@ -1,11 +1,13 @@
 This repository contains model trained to predict orientation {0:'up', 1:'down'} of images.
 The model '.pkl' file contains a dictionary with the following keys:
 
-```
+```python:
+{
 'optimizer': optimizer state dictionary
 'scheduler': scheduler state dictionary
 'model': model state dictionary
 'epoch': checkpoint epoch
+}
 ```
 
 ---