---
license: apache-2.0
tags:
  - vision
  - image-classification
  - fine-tuning
  - transformer
  - vit
datasets:
  - beans
metrics:
  - accuracy
  - f1
---

# Vision Transformer Fine-tuned on Beans

Este modelo é um **Vision Transformer (ViT)** fine-tuned no dataset **Beans** para classificação de imagens.

## 📝 Descrição do Modelo

O modelo foi treinado no dataset **Beans**, que contém imagens de folhas de feijão classificadas em **3 categorias**:

- `angular_leaf_spot` (mancha angular da folha),
- `bean_rust` (ferrugem do feijão),
- `healthy` (saudável).

A fine-tuning foi feita a partir do modelo pré-treinado `google/vit-base-patch16-224-in21k`, ajustado para 3 classes.

## 📌 Uso Pretendido

Este modelo pode ser usado para tarefas de **classificação de imagens**, ajudando na identificação de doenças em folhas de feijão.

## 🚀 Como Usar

Você pode usar o modelo diretamente com o **Hugging Face Transformers**:

```python
from transformers import pipeline

classifier = pipeline('image-classification', model='your-username/your-model-name')
results = classifier('path/to/your/image.jpg')
print(results)
```