face-emotion / README.md
gerhardien's picture
Update README.md
88e7271 verified
metadata
license: mit
base_model:
  - google/vit-base-patch16-224-in21k
pipeline_tag: image-classification
datasets:
  - Jeneral/fer-2013

Model Description

This model is a fine-tuned version of google/vit-base-patch16-224-in21k. It has been fine-tuned on the FER2013 dataset, which contains facial expression images. The model classifies images into four categories of facial expressions:

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Evaluation

Results

Results Summary Training Accuracy: 83.27% Validation Accuracy: 76.52%