import gradio as gr
from transformers import pipeline


### https://huggingface.co/blog/fine-tune-vit -->> the blog post how to fine tune ViT



pipe = pipeline(task="image-classification", 
                model="nateraw/vit-base-beans")
                
gr.Interface.from_pipeline(pipe,
                           examples=["example-leaf.jpeg"],
                           ).launch()