Update README.md

README.md

@@ -16,10 +16,10 @@ Epoch: 28 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
 to use this model you have to:
 
 1. download the model
-2. # load pretrained model resnet18
+2. load pretrained model resnet18
 3. model = models.resnet18(pretrained=True)
-4. # load checkpoint from your local
+4. load checkpoint from your local
 5. checkpoint = torch.load('butterfly_resnet_checkpoint.model')
 7. model_for_predict.load_state_dict(checkpoint)
-8. # predict the images
+8. predict the images
 9. model_for_predict.eval())