You can deploy this model and start earning money today!

aadhi3/
RoBert_Model

Name: RoBert_Model
Availability: InStock

pytorch

emotion-classification

Views

No views yet

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