Update README.md

README.md

@@ -1,6 +1,5 @@
 ---
 license: apache-2.0
-base_model: openai/whisper-tiny
 tags:
 - generated_from_trainer
 datasets:
@@ -14,10 +13,10 @@ model-index:
       name: Automatic Speech Recognition
       type: automatic-speech-recognition
     dataset:
-      name: PolyAI/minds14
+      name: minds14
       type: PolyAI/minds14
       config: en-US
-      split: train
+      split: train[450:]
       args: en-US
     metrics:
     - name: Wer
@@ -25,16 +24,18 @@ model-index:
       value: 0.3604651162790697
 ---
 
+
 <!-- 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. -->
 
 # whisper-tiny-english-us
 
-This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
+This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the minds14 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.7152
-- Wer Ortho: 0.360700
-- Wer: 0.360465
+- Wer Ortho: 0.3607
+- Wer: 0.3604
+
 
 ## Model description
 
@@ -48,6 +49,7 @@ More information needed
 
 More information needed
 
+
 ## Training procedure
 
 ### Training hyperparameters
@@ -62,11 +64,12 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_steps: 50
 - training_steps: 500
 
+
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer     |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:-------:|
-| 0.0007        | 17.86 | 500  | 0.7152          | 0.360700   | 0.360465|
+| 0.0007        | 17.86 | 500  | 0.7152          | 0.3607    | 0.3604  |
 
 
 ### Framework versions
@@ -75,3 +78,7 @@ The following hyperparameters were used during training:
 - Pytorch 2.0.1+cu118
 - Datasets 2.14.4
 - Tokenizers 0.13.3
+
+
+
+