Spaces:
Running
Running
'final_1'
Browse files
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="
|
|
|
3 |
|
4 |
pipe = pipeline(task="image-classification", model="imjeffhi/pokemon_classifier")
|
5 |
|
6 |
+
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()
|