File size: 636 Bytes
8eadbcf 7e4c6be 8eadbcf 7e4c6be 8eadbcf 7e4c6be 8eadbcf 7e4c6be |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
---
library_name: pytorch
tags:
- emotion-classification
- roberta
- deep-learning
- transformer
license: apache-2.0
---
# RoBert Emotion Classifier
PyTorch model trained for emotion classification on XYZ dataset.
Saved format: `state_dict`.
## 🔧 How to Load the Model
```python
from huggingface_hub import hf_hub_download
import torch
from BertEmotionClassifier import BertEmotionClassifier
model_path = hf_hub_download(repo_id="aadhi3/RoBert_Model", filename="model.pth")
model = BertEmotionClassifier(num_labels=5)
model.load_state_dict(torch.load(model_path, map_location="cpu"))
model.eval()
``` |