Update README.md

README.md

@@ -1,3 +1,17 @@
-This model uses distilroberta-base pretrained weights, with a classification head fine-tuned to classify news articles into 3 categories (bad, medium, good).
+---
+language: en
+datasets:
+- valurank/news-small
+---
 
-The dataset used to fine-tune the model is news-small, the 300 article news dataset manually annotated by Alex.
+# DistilROBERTA fine-tuned for news classification
+
+This model is based on [distilroberta-base](https://huggingface.co/distilroberta-base) pretrained weights, with a classification head fine-tuned to classify news articles into 3 categories (bad, medium, good).
+
+## Training data
+
+The dataset used to fine-tune the model is [news-small](https://huggingface.co/datasets/valurank/news-12factor), the 300 article news dataset manually annotated by Alex.
+
+## Inputs
+
+Similar to its base model, this model accepts inputs with a maximum length of 512 tokens.