Creating image classification model with Hugging Face