mestrevh/vit-classification-model

🏆 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)