ComenzarEmpieza gratis

Image classification

In this exercise, you will use the same flickr dataset as previously, which has 30,000 images and associated captions. Now you will classify the image using a pretrained model.

Photo of 2 people, 1 is playing the guitar

The sample image (image) and pipeline module (pipeline) have been loaded.

Este ejercicio forma parte del curso

Multi-Modal Models with Hugging Face

Ver curso

Instrucciones del ejercicio

  • Load the image-classification pipeline with google/mobilenet_v2_1.0_224 pretrained model.
  • Use the pipeline to predict the class of the sample image.
  • Print the first (highest probability) label.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# 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:", ____)
Editar y ejecutar código