Giuliano's picture
Update app.py
c861968
raw
history blame
317 Bytes
import gradio as gr
### https://huggingface.co/blog/fine-tune-vit -->> the blog post how to fine tune ViT
examples = ["example-leaf.jpeg"]
iface = gr.Interface("huggingface/nateraw/vit-base-beans",
inputs="image",
examples=examples,
outputs=gr.outputs.Label(num_top_classes=5, label='Prediction'))
iface.launch()