app.py

from transformers import pipeline
from PIL import Image

# Cargar el pipeline para clasificación de imágenes
pipe = pipeline("image-classification", model="mateoluksenberg/dit-base-Classifier_CM05")

# Ruta a la imagen que deseas clasificar dentro del contenedor
image_path = 'cm5.jpg'

# Abrir la imagen desde la ruta local y convertirla a RGB
image = Image.open(image_path).convert('RGB')

# Realizar la clasificación de la imagen
result = pipe(image)

# Imprimir el resultado
print(result)