Is it Huggable?

An image classifier that tells if something's huggable or not?

Made using fastai.

Play with it on the website or on HuggingFace Space.

How it works?

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.

Examples

Important Notes

• 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.

Contributing

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!