# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("torresflo/Poke-Model")
model = AutoModelForImageClassification.from_pretrained("torresflo/Poke-Model")Quick Links
Poké Model
Poké Model is a Pokémon Classifier created to be used with Pokédex 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.
License
Distributed under the GNU General Public License v3.0. See here for more information.
- Downloads last month
- 7
# 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")