


A vision transformer finetuned to classify the age of a given person's face. 




```python
import requests
from PIL import Image
from io import BytesIO

from transformers import ViTFeatureExtractor, ViTForImageClassification

# Get example image from official fairface repo + read it in as an image
r = requests.get('https://github.com/dchen236/FairFace/blob/master/detected_faces/race_Asian_face0.jpg?raw=true')
im = Image.open(BytesIO(r.content))

# Init model, transforms
model = ViTForImageClassification.from_pretrained('nateraw/vit-age-classifier')
transforms = ViTFeatureExtractor.from_pretrained('nateraw/vit-age-classifier')

# Transform our image and pass it through the model
inputs = transforms(im, return_tensors='pt')
output = model(**inputs)

# Predicted Class probabilities
proba = output.logits.softmax(1)

# Predicted Classes
preds = proba.argmax(1)
```
{"tags": ["image-classification", "pytorch"], "datasets": ["fairface"]}
transformers pytorch vit image-classification dataset:fairface doi:10.57967/hf/1259 autotrain_compatible endpoints_compatible has_space region:us
TAGS
#transformers #pytorch #vit #image-classification #dataset-fairface #doi-10.57967/hf/1259 #autotrain_compatible #endpoints_compatible #has_space #region-us 




A vision transformer finetuned to classify the age of a given person's face. 





TAGS
#transformers #pytorch #vit #image-classification #dataset-fairface #doi-10.57967/hf/1259 #autotrain_compatible #endpoints_compatible #has_space #region-us 

passage: TAGS
#transformers #pytorch #vit #image-classification #dataset-fairface #doi-10.57967/hf/1259 #autotrain_compatible #endpoints_compatible #has_space #region-us 
