Update README.md

README.md

@@ -9,6 +9,11 @@ metrics:
 model-index:
 - name: sentiment_analysis_model
   results: []
+datasets:
+- stanfordnlp/imdb
+language:
+- en
+new_version: tashrifmahmud/sentiment_analysis_model_v2
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,7 +21,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # sentiment_analysis_model
 
-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' dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.2366
 - Accuracy: 0.9310
@@ -59,4 +64,4 @@ The following hyperparameters were used during training:
 - Transformers 4.46.2
 - Pytorch 2.5.1+cu121
 - Datasets 3.1.0
-- Tokenizers 0.20.3
+- Tokenizers 0.20.3