Update README.md

README.md

@@ -12,14 +12,15 @@ metrics:
 model-index:
 - name: superhero-distilbert-predictor
   results: []
+datasets:
+- rlogh/superhero-texts
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
 # superhero-distilbert-predictor
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the superhero-texts dataset.
+This model maps brief descriptions of popular superheroes to their respective comic book universes. 
+
 It achieves the following results on the evaluation set:
 - Loss: 0.0161
 - Accuracy: 1.0
@@ -27,17 +28,14 @@ It achieves the following results on the evaluation set:
 - Precision: 1.0
 - Recall: 1.0
 
-## Model description
-
-More information needed
-
 ## Intended uses & limitations
 
-More information needed
+This model is strictly intended for educational use. Do not use this model to draw real world conclusions.
 
 ## Training and evaluation data
 
-More information needed
+This model was trained on an augmented set of 1100 synthetically generated superhero descriptions and their respective universe label.
+This model was validated against a set of 100 original, human curated descriptions.
 
 ## Training procedure
 
@@ -68,4 +66,4 @@ The following hyperparameters were used during training:
 - Transformers 4.56.1
 - Pytorch 2.8.0+cu126
 - Datasets 4.0.0
-- Tokenizers 0.22.0
+- Tokenizers 0.22.0