Update README.md

README.md

@@ -1,3 +1,13 @@
 ---
 license: apache-2.0
+language: 
+  - en
+datasets:
+- arxiv_dataset
+tags:
+- distilbert
 ---
+
+# DistilBERT ArXiv Category Classification
+
+DistilBERT model fine-tuned on a small subset of the [ArXiv dataset](https://www.kaggle.com/datasets/Cornell-University/arxiv) to predict the category of a given paper.