second

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import pipeline
 
-pipe = pipeline(task="image-classification", model="microsoft/beit-base-patch16-224-pt22k-ft22k")
+pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")
 
-gr.Interface.from_pipeline(pipe, title="22k Image Classification", description="Object Recognition using Microsoft BEIT", examples = ['wonder_cat.jpg', 'aki_dog.jpg',], allow_flagging="never").launch(inbrowser=True)
+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)

requirements.txt

@@ -1 +1,2 @@
-transformers
+transformers
+torch