RonanMcGovern commited on
Commit
3e08161
·
verified ·
1 Parent(s): 2b17274

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +50 -0
README.md ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ metrics:
5
+ - wer
6
+ base_model:
7
+ - openai/whisper-small
8
+ tags:
9
+ - whisper
10
+ - stt
11
+ - speech-to-text
12
+ - speech
13
+ - automatic-speech-recognition
14
+ - fine-tuned
15
+ ---
16
+ # Whisper Small Atc 5S
17
+
18
+ Fine-tuned Whisper model based on [openai/whisper-small](https://huggingface.co/openai/whisper-small).
19
+
20
+ ## Training Results
21
+
22
+ | Metric | Base Model | Fine-tuned |
23
+ |--------|------------|------------|
24
+ | WER | 55.54% | 69.23% |
25
+
26
+ **Note:** WER increased by 13.69% compared to base model
27
+
28
+ ## Training Details
29
+
30
+ - **Base Model:** [openai/whisper-small](https://huggingface.co/openai/whisper-small)
31
+ - **Training Dataset:** [Trelis/atc-train-1gb-5s](https://huggingface.co/datasets/Trelis/atc-train-1gb-5s)
32
+ - **Train Loss:** 0.3593
33
+ - **Training Time:** 3.6 minutes
34
+
35
+ ## Inference
36
+
37
+ ```python
38
+ from transformers import pipeline
39
+
40
+ asr = pipeline("automatic-speech-recognition", model="Trelis/whisper-small-atc-5s")
41
+ result = asr("path/to/audio.wav")
42
+ print(result["text"])
43
+ ```
44
+
45
+ ## Training Logs
46
+
47
+ Full training logs are available in [training_log.txt](training_log.txt).
48
+
49
+ ---
50
+ *Fine-tuned using [Trelis Studio](https://studio.trelis.com)*