Update README.md

README.md

@@ -1,12 +1,12 @@
 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:
 
----
+```
 'optimizer': optimizer state dictionary
 'scheduler': scheduler state dictionary
 'model': model state dictionary
 'epoch': checkpoint epoch
----
+```
 
 ---
 framework: pytorch
@@ -17,14 +17,11 @@ imgSize: 224x224x3
 
 The model can be initialized as:
 
-'''
-# download pretrained model
+```python:
 model = torchvision.models.efficientnet_b1(pretrained=True)
 
-# define input and output features size
 in_features = model.classifier[1].in_features
 out_features = 2
 
-# replace the last layer
 model.classifier[1] = nn.Linear(in_features, out_features, bias=True)
-'''
+```