alyxx
update
149fff5
raw
history blame contribute delete
502 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification",
model="kaiku03/vit-base-1stGen-Pokemon-Images")
gr.Interface.from_pipeline(pipe,
title="Whos's that pokemon",
description="Pokemon Image Classification",
examples = ['bulba.jfif', 'charmander.png', 'pika.jfif'],
article = "Author: kaiku",
).launch()