CoolFace
Modelpublic

Wadden/vit-beans-demo

sourceHugging Faceapache-2.0updated 18d agoView on Hugging Face
0likes84downloads
Model Card

ViT Beans Classifier

Fine-tuned google/vit-base-patch16-224-in21k on the official beans dataset for 4 epochs. It classifies bean leaves as angular_leaf_spot, bean_rust, or healthy.

Test result

MetricValue
Accuracy0.9609

Usage

from transformers import pipeline classifier = pipeline("image-classification", model="Wadden/vit-beans-demo") classifier("leaf.jpg")

Interactive demo

The quantized ONNX export is executed directly in the browser with Transformers.js:

[Open the interactive Space](https://huggingface.co/spaces/Wadden/vit-beans-space)