Update README.md

Added tags and clarified dataset

README.md

@@ -11,6 +11,10 @@ metrics:
 model-index:
 - name: distilbert-base-uncased-imdb
   results: []
+language:
+- en
+library_name: transformers
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -18,7 +22,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-imdb
 
-This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased)
+on [Imdb Movie Reviews Sentiment Analysis (Kaggle)](https://www.kaggle.com/code/gauravduttakiit/imdb-movie-reviews-sentiment-analysis/data) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.2898
 - Accuracy: 0.9284
@@ -66,4 +71,4 @@ The following hyperparameters were used during training:
 - Transformers 4.44.0
 - Pytorch 2.4.0+cu121
 - Datasets 2.21.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1