An image classifier that tells if something's huggable or not?
Made using fastai.
Play with it on the website or on HuggingFace Space.
A pretrained image classification model (here resnet34) is fine tuned on images of multiple examples of both categories, like for huggable, pillow photos and for not huggable, images of cactus.
For steps to make the model check the model notebook in the repo.
- Huggability is very subjective.
- The model is not looking for particular faces or characters but general features of it.
- It's just picking up on features that makes a thing huggable or vice-versa from the training data, like sharpness.
If you want to contribute code, just write a quick pull request and the developers will take a look at it. If you want to suggest an idea, just write an issue and the developers will check it out!