Downloads · 30 days
7
39% of all-time downloads
gecko-classifier-model
jancarloonce/gecko-classifier-model
gecko-classifier-model is a machine learning model from jancarloonce. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This is a fine-tuned Vision Transformer (ViT) model that classifies geckos into:
All-time downloads
18
Public
Parameters
85.8M
4.5 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt2.1 GB · 55%
At a glance
- Model type
- vit
- Access
- Public
- Created
- Jun 1, 2025
- Updated
- Jun 1, 2025
- SHA
- 9c487f4c
Model card
From the Hugging Face model README
Gecko Species Classifier 🦎
This is a fine-tuned Vision Transformer (ViT) model that classifies geckos into:
- Leopard Gecko
- Fat-Tail Gecko
Model Details
- Base model:
google/vit-base-patch16-224 - Fine-tuned using
transformersanddatasetson a custom dataset hosted here
Usage
from transformers import AutoImageProcessor, ViTForImageClassification
from PIL import Image
import torch
model = ViTForImageClassification.from_pretrained("jancarloonce/gecko-classifier-model")
processor = AutoImageProcessor.from_pretrained("jancarloonce/gecko-classifier-model")
image = Image.open("your_image.jpg")
inputs = processor(images=image, return_tensors="pt")
outputs = model(**inputs)
pred = outputs.logits.argmax(-1).item()
label = model.config.id2label[pred]
print(f"Predicted: {label}")
- Type
- vit
- Created
- Jun 1, 2025
- Updated
- Jun 1, 2025