| import gradio as gr | |
| from transformers import pipeline | |
| pipe = pipeline(task="image-classification", | |
| # model that can do 22k-category classification | |
| model="microsoft/beit-base-patch16-224-pt22k-ft22k") | |
| gr.Interface.from_pipeline(pipe, | |
| title="22k Image Classification", | |
| description="Object Recognition using Microsoft BEIT (22k classes)", | |
| examples = ['data/wonder_cat.jpg', 'data/aki_dog.jpg',], | |
| article = "Author: <a href=\"https://huggingface.co/rowel\">Rowel Atienza</a>", | |
| ).launch() |