Aplicaciones de machine learning

Gradio, una biblioteca rápida y sencilla para crear y compartir aplicaciones de aprendizaje automático, está integrada con Pipeline para crear rápidamente una interfaz simple para inferencia.

Antes de comenzar, asegúrate de que Gradio esté instalado.

!pip install gradio

Crea un pipeline para tu tarea y, a continuación, pásalo a la función Interface.from_pipeline de Gradio para crear la interfaz. Gradio determina automáticamente los componentes de entrada y de salida adecuados para un Pipeline.

Agrega launch para crear un servidor web e iniciar la aplicación.

from transformers import pipeline
import gradio as gr

pipeline = pipeline("image-classification", model="google/vit-base-patch16-224")
gr.Interface.from_pipeline(pipeline).launch()

La aplicación web se ejecuta en un servidor local de forma predeterminada. Para compartir la aplicación con otros usuarios, establezca share=True en launch para generar un enlace público temporal. Para una solución más permanente, aloje la aplicación en Spaces de Hugging Face.

gr.Interface.from_pipeline(pipeline).launch(share=True)

El Space de abajo se crea con el código anterior y se aloja en Spaces.