Update README.md

README.md

@@ -6,6 +6,13 @@ tags:
 model-index:
 - name: distilbert-base-uncased-finetuned-imdb
   results: []
+datasets:
+- stanfordnlp/imdb
+language:
+- en
+metrics:
+- perplexity
+library_name: transformers
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -19,15 +26,15 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+Fine-tuned distilbert for masked language modeling
 
 ## Intended uses & limitations
 
-More information needed
+Open source
 
 ## Training and evaluation data
 
-More information needed
+imdb dataset
 
 ## Training procedure
 
@@ -57,4 +64,4 @@ The following hyperparameters were used during training:
 - Transformers 4.40.2
 - Pytorch 2.2.1+cu121
 - Datasets 2.19.1
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1