Downloads · 30 days
5
13% of all-time downloads
emotion-vit-base
sfuneno/emotion-vit-base
emotion-vit-base is a machine learning model from sfuneno. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository contains a fine-tuned Google Vision Transformer (ViT-Base) model for image classification. The model is based on google/vit-base-patch16-224 and has been tuned on a task-specific dataset to make it per…
All-time downloads
40
Public
Parameters
85.8M
343 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors343 MB · 100%
At a glance
- Model type
- vit
- Access
- Public
- Created
- Dec 19, 2025
- Updated
- Dec 19, 2025
- SHA
- 91630d1b
Model card
From the Hugging Face model README
Tuned Vision Transformer (ViT-Base)
Overview
This repository contains a fine-tuned Google Vision Transformer (ViT-Base) model for image classification.
The model is based on google/vit-base-patch16-224 and has been tuned on a task-specific dataset to make it perform well in emotion classification.
Model Details
- Base model:
google/vit-base-patch16-224 - Architecture: Vision Transformer (ViT)
- Patch size: 16 × 16
- Input resolution: 224 × 224
- Framework: PyTorch + Hugging Face Transformers
- Fine-tuning: Supervised fine-tuning on labeled RAF-DB dataset
Usage
Load the model
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained(
"sfuneno/google-vit-base"
)
model = AutoModelForImageClassification.from_pretrained(
"sfuneno/google-vit-base"
)
- Type
- vit
- Created
- Dec 19, 2025
- Updated
- Dec 19, 2025