Update README.md

README.md

@@ -26,7 +26,7 @@ Epoch: 28 Train Loss: 0.17 Train Accuracy: 0.96 Test Accuracy: 0.90
 2. load pretrained model resnet18
 3. model_for_predict = models.resnet18(pretrained=True)
 4. load checkpoint from your local
-5. checkpoint = torch.load('butterfly_resnet_checkpoint.model')
+5. checkpoint = torch.load('pytorch_model.bin')
 7. model_for_predict.load_state_dict(checkpoint)
 8. predict the images
 9. model_for_predict.eval())