Update README.md

README.md

@@ -9,6 +9,8 @@ metrics:
 model-index:
 - name: bert-paper-classifier-arxiv
   results: []
+datasets:
+- arxiv-community/arxiv_dataset
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,24 +18,18 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert-paper-classifier-arxiv
 
-This model is a fine-tuned version of [microsoft/deberta-v3-base](https://huggingface.co/microsoft/deberta-v3-base) on the None dataset.
+This model is a fine-tuned version of [microsoft/deberta-v3-base](https://huggingface.co/microsoft/deberta-v3-base) on the [aXiv dataset](https://www.kaggle.com/datasets/Cornell-University/arxiv).
 It achieves the following results on the evaluation set:
 - Loss: 1.1651
 - Accuracy: 0.6854
 
 ## Model description
 
-More information needed
-
-## Intended uses & limitations
-
-More information needed
+Predict category of the paper based on its title and article.
 
 ## Training and evaluation data
 
-More information needed
-
-## Training procedure
+[aXiv dataset](https://www.kaggle.com/datasets/Cornell-University/arxiv)
 
 ### Training hyperparameters
 
@@ -65,4 +61,4 @@ No additional optimizer arguments
 - Transformers 4.50.3
 - Pytorch 2.6.0+cu124
 - Datasets 3.5.0
-- Tokenizers 0.21.1
+- Tokenizers 0.21.1