gbryan commited on
Commit
9ef845e
1 Parent(s): ef95e07
Files changed (4) hide show
  1. app.py +1 -1
  2. arceus.jpg +0 -0
  3. charizard.png +0 -0
  4. rayquaza.png +0 -0
app.py CHANGED
@@ -3,4 +3,4 @@ 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)
 
3
 
4
  pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")
5
 
6
+ gr.Interface.from_pipeline(pipe, title="Pokemon Classifier", description="Pokemon Classifier using a fine-tuned version of ViT-base on a collected set of Pokémon images", examples = ['arceus.jpg', 'charizard.png', 'rayquaza.png'], allow_flagging="never").launch()
arceus.jpg ADDED
charizard.png ADDED
rayquaza.png ADDED