Giuliano's picture
Update app.py
5a8c331
raw
history blame contribute delete
No virus
385 Bytes
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()