File size: 436 Bytes
e2804ec e3b2a28 e2804ec e3b2a28 e2804ec |
1 2 3 4 5 6 7 8 9 10 11 12 |
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification",
# model that can do 22k-category classification
model="huggingface/google/vit-base-patch16-224-in21k")
gr.Interface.load(pipe,
theme="default",
examples=[["alligator.jpg"], ["laptop.jpg"]],
css=".footer{display:none !important}",
title=None).launch() |