gbryan's picture
s
c9b3fda
raw
history blame
369 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification", model="microsoft/beit-base-patch16-224-pt22k-ft22k")
gr.Interface.from_pipeline(pipe, title="22k Image Classification", description="Object Recognition using Microsoft BEIT", examples = ['wonder_cat.jpg', 'aki_dog.jpg',], allow_flagging="never").launch(inbrowser=True)