Upload folder using huggingface_hub

README.md

@@ -1,17 +1,27 @@
+---
+library_name: pytorch
+tags:
+- emotion-classification
+- roberta
+- deep-learning
+- transformer
+license: apache-2.0
+---
+
 # RoBert Emotion Classifier
-PyTorch model trained for classification on XYZ dataset.
-Save format: state_dict
-To load: import Model_class and load state dict.
 
-```py
+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 Model_class import MyClassifier  # architecture from HF repo cloned or imported manually
+from BertEmotionClassifier import BertEmotionClassifier
 
-# Download model weights from HuggingFace
 model_path = hf_hub_download(repo_id="aadhi3/RoBert_Model", filename="model.pth")
 
-model = MyClassifier(num_labels=5)
+model = BertEmotionClassifier(num_labels=5)
 model.load_state_dict(torch.load(model_path, map_location="cpu"))
 model.eval()
 ```