import gradio as gr

titulo = "Mi primer demo con Hugging Face"
descripcion = "Este es un demo ejecutado durante la clase con Platzi."

# Cargar el modelo desde Hugging Face
modelo = gr.load("huggingface/microsoft/swin-tiny-patch4-window7-224")

# Verificar la funcionalidad del modelo cargado
def predict(image):
    return modelo(image)

# Crear una nueva interfaz reutilizando el modelo
demo = gr.Interface(
    fn=predict,                       # Usar la función de predicción
    inputs=gr.Image(label="Carga una imagen aquí"),  # Input definido
    outputs=gr.Label(num_top_classes=3), # Outputs del modelo cargado
    title=titulo,                    # Título de la interfaz
    description=descripcion           # Descripción de la interfaz
)

# Lanzar la aplicación
demo.launch()