SoulPerforms
/

Butterfly_image_classification_resnet18

Image Classification

Model card Files Files and versions

SoulPerforms commited on Feb 25, 2024

Commit

8c27fdd

·

verified ·

1 Parent(s): 11a41f4

Update README.md

Files changed (1) hide show

README.md +5 -1

README.md CHANGED Viewed

@@ -22,4 +22,8 @@ to use this model you have to:
 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())

 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())
+![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/65bdde53259bc6caeb094e85/b__Bo6f-R0el__5yqusRc.jpeg)