metadata
tags:
- vision
- image-classification
- transformers
datasets:
- beans
license: apache-2.0
🏆 Modelo de Classificação de Imagens - Vision Transformer (ViT)
Este modelo foi treinado para classificação de imagens usando o Vision Transformer (ViT).
📌 Repositório do Space (Aplicação Web):
👉 mestrevh/image-classification-vit
📌 Descrição do Modelo
Este modelo foi treinado para classificar imagens em diferentes categorias com base no dataset Beans.
🔹 Modelo Base: google/vit-base-patch16-224-in21k
🔹 Dataset utilizado: beans
🔹 Tarefa: Classificação de imagens
🚀 Como Usar
Você pode carregar este modelo diretamente com transformers:
from transformers import pipeline
# Carregar modelo treinado do Hugging Face Hub
model_id = "mestrevh/NOME_DO_MODELO"
classifier = pipeline("image-classification", model=model_id)
# Fazer uma predição
image_path = "caminho/para/sua/imagem.jpg"
result = classifier(image_path)
print(result)