Update README.md

README.md

@@ -10,4 +10,16 @@ Butterfly image classification model that use pre-trained cnn model resnet18 and
 The model used the best checkpoint with 90% test accuracy. The model constructed on Pytorch environment.
 
 training and testing result:
-Epoch: 28 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
+
+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
+3. model = models.resnet18(pretrained=True)
+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
+9. model_for_predict.eval())