diff --git a/README.md b/README.md index 4ebfdc8..f57c389 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,21 @@ -Run emotion model with 3 lines of code on single text example using Hugging Face's pipeline command: +## Description + +This deep neural network classifies the emotions in English text data. The model was trained on diverse datasets and predicts 7 emotions: + +1) anger +2) disgust +3) fear +4) joy +5) neutral +6) sadness +7) surprise + +## Application + +a) Run emotion model with 3 lines of code on single text example using Hugging Face's pipeline command: [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/j-hartmann/emotion-english-distilroberta-base/blob/main/simple_emotion_pipeline.ipynb) -Run emotion model on multiple examples and full datasets (e.g., .csv files): +b) Run emotion model on multiple examples and full datasets (e.g., .csv files): [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/j-hartmann/emotion-english-distilroberta-base/blob/main/emotion_prediction_example.ipynb) \ No newline at end of file