mestrevh commited on
Commit
db228f2
·
verified ·
1 Parent(s): 916357c

Create Readme

Browse files
Files changed (1) hide show
  1. README.md +42 -0
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - vision
4
+ - image-classification
5
+ - transformers
6
+ datasets:
7
+ - beans
8
+ license: apache-2.0
9
+ ---
10
+
11
+ # 🏆 Modelo de Classificação de Imagens - Vision Transformer (ViT)
12
+
13
+ Este modelo foi treinado para **classificação de imagens** usando o **Vision Transformer (ViT)**.
14
+
15
+ 📌 **Repositório do Space (Aplicação Web)**:
16
+ 👉 [mestrevh/image-classification-vit](https://huggingface.co/spaces/mestrevh/image-classification-vit)
17
+
18
+ ---
19
+
20
+ ## 📌 **Descrição do Modelo**
21
+ Este modelo foi treinado para classificar imagens em diferentes categorias com base no dataset **Beans**.
22
+
23
+ 🔹 **Modelo Base**: [`google/vit-base-patch16-224-in21k`](https://huggingface.co/google/vit-base-patch16-224-in21k)
24
+ 🔹 **Dataset utilizado**: [`beans`](https://huggingface.co/datasets/beans)
25
+ 🔹 **Tarefa**: Classificação de imagens
26
+
27
+ ---
28
+
29
+ ## 🚀 **Como Usar**
30
+ Você pode carregar este modelo diretamente com `transformers`:
31
+
32
+ ```python
33
+ from transformers import pipeline
34
+
35
+ # Carregar modelo treinado do Hugging Face Hub
36
+ model_id = "mestrevh/NOME_DO_MODELO"
37
+ classifier = pipeline("image-classification", model=model_id)
38
+
39
+ # Fazer uma predição
40
+ image_path = "caminho/para/sua/imagem.jpg"
41
+ result = classifier(image_path)
42
+ print(result)