varadch commited on
Commit
9da3969
·
verified ·
1 Parent(s): af6bee6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -3
README.md CHANGED
@@ -1,3 +1,19 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """# EfficientNet-B2 Emotion Detection (v1)
2
+
3
+ This model classifies facial emotions into 7 categories:
4
+ `angry`, `disgust`, `fear`, `happy`, `neutral`, `sad`, `surprise`.
5
+
6
+ **Architecture:** EfficientNet-B2
7
+ **Training Platform:** Kaggle GPU
8
+ **Accuracy:** ~80.25%
9
+ **Framework:** PyTorch
10
+ **Developer:** Varad V. Choudhari (Atman AI)
11
+ **License:** MIT
12
+
13
+ ### Example Usage
14
+ ```python
15
+ from huggingface_hub import hf_hub_download
16
+ path = hf_hub_download(
17
+ repo_id="AtmanAI/emotion-detection-efficientnet-b2-v1",
18
+ filename="efficientnet_b2_emotion_final.pth"
19
+ )