| 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() | |
| ``` |