Instructions to use torresflo/Poke-Model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use torresflo/Poke-Model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="torresflo/Poke-Model") 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("torresflo/Poke-Model") model = AutoModelForImageClassification.from_pretrained("torresflo/Poke-Model", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 930 Bytes
761a76d b7bd686 761a76d b7bd686 761a76d b7bd686 761a76d b9adeb2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | ---
license: gpl-3.0
tags:
- vision
- image-classification
- Pokémon
widget:
- src: https://huggingface.co/torresflo/Poke-Model/resolve/main/examples/1.jpg
example_title: Bulbasaur
- src: https://huggingface.co/torresflo/Poke-Model/resolve/main/examples/2.jpg
example_title: Charizard
- src: https://huggingface.co/torresflo/Poke-Model/resolve/main/examples/3.jpg
example_title: Blastoise
---
# Poké Model
Poké Model is a Pokémon Classifier created to be used with [Pokédex AI](https://github.com/torresflo/Pokedex-AI). It is a fine-tuned model of google/vit-base-patch16-224 to classify Pokémon of the first generation.
More information on how to generate and how to use the model can be found on this [dedicated repository](https://github.com/torresflo/Poke-Model).
## License
Distributed under the GNU General Public License v3.0. See [here](https://www.gnu.org/licenses/gpl-3.0.en.html) for more information.
|