📝 add to readme

README.md

@@ -7,3 +7,34 @@ datasets:
 ---
 
 # ViT For Age Classification
+
+A vision transformer finetuned to classify the age of a given person's face. 
+
+
+## Usage in Transformers
+
+```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)
+```