gbryan commited on
Commit
842b256
1 Parent(s): d7289bc
Files changed (1) hide show
  1. app.py +6 -0
app.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")
5
+
6
+ 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)