File size: 689 Bytes
a294d9c 8bfaa9e a294d9c |
1 2 3 4 5 6 7 8 9 10 11 12 |
import gradio as gr
from transformers import pipeline
vision_classifier = pipeline(task="image-classification",
model="microsoft/beit-base-patch16-224-pt22k-ft22k")
iface = gr.Interface.from_pipeline(vision_classifier,
title="2-line Object Recognition using HF Pipeline",
description="Demonstrates how to use a pipeline and model hub.",
examples=['data/wonder_cat.jpg', 'data/aki_dog.jpg',],
article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>",
).launch()
|