Update README.md

README.md

@@ -1,11 +1,14 @@
+---
+pipeline_tag: text-classification
+---
 # Model Card for [natong19/moralization_classifier](https://huggingface.co/natong19/moralization_classifier)
 
 A classifer for detecting moralizations, soft refusals and unsolicited advice.
 
 Trained on [OpenLeecher/lmsys_chat_1m_clean](https://huggingface.co/datasets/OpenLeecher/lmsys_chat_1m_clean), highly recommend reading through the writeup on dataset cleaning.
 
-### Example usage:
-```
+### Quickstart
+```python
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
@@ -83,7 +86,7 @@ if __name__ == "__main__":
 
 ```
 
-### Evaluation results:
+### Evaluation results
 - eval_loss: 0.0844
 - eval_accuracy: 0.9800
 - eval_f1: 0.9841