You can deploy this model and start earning money today!

jancarloonce/
gecko-classifier-model

Name: gecko-classifier-model
Availability: InStock

safetensors

vit

Views

No views yet

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

python

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}")