diff --git a/README.md b/README.md index f57c389..e9d0485 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,6 @@ ## Description -This deep neural network classifies the emotions in English text data. The model was trained on diverse datasets and predicts 7 emotions: +With this model, you can classify emotions in English text data. The model was trained on diverse datasets and predicts 7 emotions: 1) anger 2) disgust @@ -18,4 +18,8 @@ a) Run emotion model with 3 lines of code on single text example using Hugging F 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 +[![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) + +## Contact + +Please reach out to jochen.hartmann@uni-hamburg.de if you have any questions or feedback. \ No newline at end of file