Returns person celebrity-style attractiveness (0 to 1) based on facial image with about 83% accuracy.

See https://www.kaggle.com/code/dima806/attractive-faces-celebs-detection-vit for more details.

Classification report:

                precision    recall  f1-score   support

    attractive     0.8297    0.8502    0.8398      5192
not attractive     0.8464    0.8255    0.8358      5192

      accuracy                         0.8378     10384
     macro avg     0.8380    0.8378    0.8378     10384
  weighted avg     0.8380    0.8378    0.8378     10384

Downloads last month: 134

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Examples

Image Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for dima806/attractive_faces_celebs_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1668)

this model

dima806
/

attractive_faces_celebs_detection

Model tree for dima806/attractive_faces_celebs_detection

Space using dima806/attractive_faces_celebs_detection 1