img_rec_VIT / app.py
teamtom's picture
updated app.py for gradio v4
9abf204
raw
history blame
248 Bytes
import gradio as gr
examples = [
'ladybug.jpg',
'revolver.jpg',
'automotive.jpg',
'frog01.jpg',
'eagle.jpg'
]
# works w/ gradio v4
demo = gr.load("google/vit-base-patch16-224", src="models", examples=examples)
demo.launch()