Update README.md

README.md

@@ -2,21 +2,22 @@
 library_name: transformers
 language: en
 license: apache-2.0
-datasets: []
+datasets:
+- stanfordnlp/imdb
 base_model:
 - google-bert/bert-base-uncased
 ---
 
-# BERT Fine-Tuned on <Dataset>
+# BERT Fine-Tuned on IMDb
 
-A fine-tuned BERT model using the <Dataset> dataset.
+A fine-tuned BERT model using the IMDb dataset.
 
 ## Model Details
 
 ### Description
 
 This model is based on the [BERT base (uncased)](https://huggingface.co/google-bert/bert-base-uncased)
-architecture and has been fine-tuned on the <Dataset> dataset.
+architecture and has been fine-tuned on the [IMDb](https://huggingface.co/datasets/stanfordnlp/imdb) dataset.
 
 - **Developed by:** [Cesar Gonzalez-Gutierrez](https://ceguel.es)
 - **Funded by:** [ERC](https://erc.europa.eu)
@@ -58,8 +59,8 @@ on the entire training dataset.
 
 ### Training Data
 
-The model was trained on the <Dataset> training partition, with validation performed on
-either the dataset’s development set (if available) or a random 20% split of the training data.
+The model was trained on the IMDb training partition, with validation performed on
+a random 20% split of the training data.
 
 #### Training Hyperparameters
 
@@ -71,7 +72,7 @@ either the dataset’s development set (if available) or a random 20% split of t
 
 ## Uses
 
-This model can be used for classification tasks aligned with the structure and intent of the <Dataset> corpus.
+This model can be used for classification tasks aligned with the structure and intent of the IMDb corpus.
 
 For broader guidance, refer to the BERT base model’s [Inteded Uses & Limitations](https://huggingface.co/google-bert/bert-base-uncased#intended-uses--limitations).
 
@@ -80,7 +81,7 @@ For broader guidance, refer to the BERT base model’s [Inteded Uses & Limitatio
 This model inherits the potential risks and limitations of its base model. For more details,
 refer to the [Limitations and bias](https://huggingface.co/google-bert/bert-base-uncased#limitations-and-bias) section of the original model documentation. 
 
-Additionally, it may reflect or amplify patterns and biases present in the <Dataset> training data.
+Additionally, it may reflect or amplify patterns and biases present in the IMDb training data.
 
 ## Hardware
 
@@ -91,4 +92,4 @@ Additionally, it may reflect or amplify patterns and biases present in the <Data
 ## Citation
 
 If you use this model in your research, please cite both the base BERT model
-and the <Dataset> source (to be added).
+and the IMDb source.