app.py

import gradio as gr
from transformers import pipeline

pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")

gr.Interface.from_pipeline(pipe, title="Pokemon Classifier", description="A fine-tuned version of ViT-base on a collected set of Pokémon images", allow_flagging="never").launch(inbrowser=True)