img_rec_VIT / app.py
teamtom's picture
updating gradio version in readme.md
ce157a4
raw
history blame
339 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) # with examples
# demo = gr.load("google/vit-base-patch16-224", src="models") # no examples
demo.launch()