File size: 174 Bytes
f6d69d3
 
 
 
 
 
1
2
3
4
5
6
from transformers import pipeline
import gradio as gr

pipe = pipeline("image-classification", model="google/vit-base-patch16-224")

gr.Interface.from_pipeline(pipe).launch()