Update README.md

README.md

@@ -1,3 +1,22 @@
----
-license: mit
----
+---
+license: mit
+language:
+- en
+base_model:
+- google-bert/bert-base-uncased
+pipeline_tag: text-classification
+tags:
+- Text_based_emotion_classification
+---
+
+# BERT-based-Emotion-Classification
+- BERT-based Emotion Classification.
+- A custom Pytorch model which uses BERT for multiclass emotion classification.
+- There are 8 different classes. Labels: `anxiety`, `bipolar`, `depression`, `fear`, `grateful`, `happy`, `sad`, `stress`
+- Version 1: Achieved an accuracy of 88%
+
+### Model Huggingface Repo
+- BERT-based-Emotion-Classification-v1: https://huggingface.co/dkp2701/BERT-based-Multiclass-Emotion-Classification
+
+### Dataset
+- https://www.kaggle.com/datasets/deepakprajapati07/emotion-detection/data