Image classification model in 5 lines with Hugging Face