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