README updated

README.md

@@ -2,4 +2,12 @@
 license: cc-by-nc-sa-4.0
 ---
 
-The current version of code is based on https://huggingface.co/nateraw/bert-base-uncased-emotion and will be improved gradually.
+Notebook: https://colab.research.google.com/drive/10ZCFvlf2UV3FjU4ymf4OoipQvqHbIItG?usp=sharing
+
+
+## Example
+
+Run `example.py`
+
+## Train
+The model is already trained on `bert-base-uncased` with the [emotions dataset](https://huggingface.co/datasets/emotion). However, you can change parameters and re-fine-tune the model by running `finetune-emotions.py`.