update model card README.md

README.md

@@ -10,7 +10,7 @@ datasets:
 metrics:
 - wer
 model-index:
-- name: Whisper base ID - Augmented
+- name: Whisper small ID - Augmented
   results:
   - task:
       name: Automatic Speech Recognition
@@ -24,18 +24,18 @@ model-index:
     metrics:
     - name: Wer
       type: wer
-      value: 11.20271950074837
+      value: 5.697759963469393
 ---
 
 <!-- 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 base ID - Augmented
+# Whisper small ID - Augmented
 
-This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the mozilla-foundation/common_voice_12_0 dataset.
+This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the mozilla-foundation/common_voice_12_0 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2277
-- Wer: 11.2027
+- Loss: 0.1483
+- Wer: 5.6978
 
 ## Model description
 
@@ -68,13 +68,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Wer     |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|
-| 0.5509        | 0.77  | 200  | 0.3243          | 13.8588 |
-| 0.3844        | 1.54  | 400  | 0.2590          | 11.1782 |
-| 0.259         | 2.31  | 600  | 0.2422          | 18.6813 |
-| 0.1608        | 3.07  | 800  | 0.2361          | 11.6678 |
-| 0.1205        | 3.84  | 1000 | 0.2277          | 11.2027 |
+| Training Loss | Epoch | Step | Validation Loss | Wer    |
+|:-------------:|:-----:|:----:|:---------------:|:------:|
+| 0.3979        | 0.77  | 200  | 0.2141          | 9.7635 |
+| 0.2625        | 1.54  | 400  | 0.1705          | 7.3636 |
+| 0.1398        | 2.31  | 600  | 0.1551          | 6.9636 |
+| 0.0666        | 3.07  | 800  | 0.1678          | 6.2043 |
+| 0.0411        | 3.84  | 1000 | 0.1483          | 5.6978 |
 
 
 ### Framework versions