jancarloonce/gecko-classifier-model

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 transformers and datasets on 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}")