Edit model card

Returns whether the person is a smoker based on image.

See https://www.kaggle.com/code/dima806/smoker-image-detection-vit for more details.

Classification report:

              precision    recall  f1-score   support

  notsmoking     0.9907    0.9464    0.9680       112
     smoking     0.9487    0.9911    0.9694       112

    accuracy                         0.9688       224
   macro avg     0.9697    0.9688    0.9687       224
weighted avg     0.9697    0.9688    0.9687       224
Downloads last month
11
Safetensors
Model size
85.8M params
Tensor type
F32
·