Classificação de imagens
Neste exercício, você vai usar o mesmo conjunto de dados do Flickr que usou antes, que tem 30.000 imagens e legendas associadas. Agora você vai classificar a imagem usando um modelo pré-treinado.

A imagem de exemplo (image) e o módulo do pipeline (pipeline) foram carregados.
Este exercício faz parte do curso
Modelos multimodais com Hugging Face
Instruções do exercício
- Carregue o pipeline
image-classificationcom o modelo pré-treinadogoogle/mobilenet_v2_1.0_224. - Use o pipeline para prever a classe da imagem de amostra.
- Imprima o primeiro (com maior probabilidade)
label.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Load the image-classification pipeline
pipe = ____
# Use the pipeline to predict the class of the sample image
pred = ____
# Print the first (highest probability) label
print("Predicted class:", ____)