Spaces:
Runtime error
Runtime error
File size: 497 Bytes
c9b3fda 842b256 ef95e07 842b256 4ae3794 |
1 2 3 4 5 6 |
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")
gr.Interface.from_pipeline(pipe, title="Pokémon Classifier", description="Pokémon image classifier using a fine-tuned version of ViT-base on a collected set of Pokémon images. To use this space, just drag and drop, or upload any Pokémon image to know its name.", examples = ['arceus.jpg', 'charizard.png', 'rayquaza.png'], allow_flagging="never").launch() |