Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,44 @@
+---
+tags:
+- image-classification
+- pytorch
+- huggingface
+- vit
+- emotion-recognition
+datasets:
+- zenodo
+base_model: trpakov/vit-face-expression
+library_name: transformers
+---
+
+# ViT Face Expression (Fine-tuned on Zenodo Dataset)
+
+This model is a fine-tuned version of [trpakov/vit-face-expression](https://huggingface.co/trpakov/vit-face-expression) on the [IFEED dataset (Zenodo Record 7963451)](https://zenodo.org/record/7963451).
+
+## Model Description
+- **Architecture**: Vision Transformer (ViT)
+- **Task**: Facial Emotion Recognition
+- **Emotions**: Angry, Disgust, Fear, Happy, Neutral, Sad, Surprise
+
+## Usage
+
+```python
+from transformers import ViTImageProcessor, ViTForImageClassification
+from PIL import Image
+import requests
+
+url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
+image = Image.open(requests.get(url, stream=True).raw)
+
+repo_name = "michaelgathara/vit-face-zenodo"
+
+processor = ViTImageProcessor.from_pretrained(repo_name)
+model = ViTForImageClassification.from_pretrained(repo_name)
+
+inputs = processor(images=image, return_tensors="pt")
+outputs = model(**inputs)
+logits = outputs.logits
+# model predicts one of the 7 emotions
+predicted_class_idx = logits.argmax(-1).item()
+print("Predicted class:", model.config.id2label[predicted_class_idx])
+```