import gradio as gr | |
from transformers import pipeline | |
pipe = pipeline(task="image-classification", | |
# model for 22k-category classification | |
model="microsoft/beit-base-patch16-224-pt22k-ft22k") | |
gr.Interface.from_pipeline(pipe, | |
title="22k Image Classification", | |
description="Object Recognition ...", | |
examples = ['wonder_cat.jpg', | |
'aki_dog.jpg',], | |
).launch(inbrowser=True) |