Image Classification
Transformers
PyTorch
TensorBoard
English
vit
Generated from Trainer
Eval Results (legacy)
Instructions to use stbnlen/pokemon_classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use stbnlen/pokemon_classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="stbnlen/pokemon_classifier") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("stbnlen/pokemon_classifier") model = AutoModelForImageClassification.from_pretrained("stbnlen/pokemon_classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
-
Adding `safetensors` variant of this model
This is an automated PR created with https://huggingface.co/spaces/safetensors/convert This new file is equivalent to `pytorch_model.bin` but safe in the sense that no arbitrary code can be put into it. These files also happen to load much faster than their pytorch counterpart: https://colab.research.google.com/github/huggingface/notebooks/blob/main/safetensors_doc/en/speed.ipynb The widgets on your model page will run using this model even if this is not merged making sure the file actually works. If you find any issues: please report here: https://huggingface.co/spaces/safetensors/convert/discussions Feel free to ignore this PR.e1f0ea1 -
update model card README.md
1b5344f -
Model save
a9f842a -
Training in progress, step 2000
fad2aa9 -
Training in progress, step 1500
12f939b -
Training in progress, step 1000
2c049e7 -
Training in progress, step 500
dda64d4
-
Training in progress, step 1000
7a8862e -
Training in progress, step 500
1219658 -
update model card README.md
48e4a62 -
Model save
4edb1ef -
initial commit
198e83d