File size: 385 Bytes
c861968
5d00db4
c861968
 
 
 
 
 
5d00db4
 
 
5a8c331
5d00db4
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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()