update model card README.md

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
+datasets:
+- imdb
 model-index:
 - name: distilbert-base-uncased-finetuned-imdb
   results: []
@@ -13,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-finetuned-imdb
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
 It achieves the following results on the evaluation set:
 - eval_loss: 3.0882
-- eval_runtime: 393.1782
-- eval_samples_per_second: 2.543
-- eval_steps_per_second: 0.041
+- eval_runtime: 385.1294
+- eval_samples_per_second: 2.597
+- eval_steps_per_second: 0.042
 - step: 0
 
 ## Model description
@@ -50,4 +52,5 @@ The following hyperparameters were used during training:
 
 - Transformers 4.31.0
 - Pytorch 2.0.1+cu118
+- Datasets 2.13.1
 - Tokenizers 0.13.3