corygong commited on
Commit
c0679d6
·
verified ·
1 Parent(s): 16444ca

Upload config

Browse files
Files changed (1) hide show
  1. README.md +110 -0
README.md ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: cc-by-4.0
5
+ library_name: nemo
6
+ datasets:
7
+ - librispeech_asr
8
+ - mozilla-foundation/common_voice_7_0
9
+ - vctk
10
+ - fisher_corpus
11
+ - Switchboard-1
12
+ - WSJ-0
13
+ - WSJ-1
14
+ - National-Singapore-Corpus-Part-1
15
+ - National-Singapore-Corpus-Part-6
16
+ thumbnail: null
17
+ tags:
18
+ - automatic-speech-recognition
19
+ - speech
20
+ - audio
21
+ - CTC
22
+ - Conformer
23
+ - Transformer
24
+ - NeMo
25
+ - pytorch
26
+ model-index:
27
+ - name: stt_en_conformer_ctc_small
28
+ results: []
29
+
30
+ ---
31
+
32
+
33
+ ## Model Overview
34
+
35
+ <DESCRIBE IN ONE LINE THE MODEL AND ITS USE>
36
+
37
+ ## NVIDIA NeMo: Training
38
+
39
+ To train, fine-tune or play with the model you will need to install [NVIDIA NeMo](https://github.com/NVIDIA/NeMo). We recommend you install it after you've installed latest Pytorch version.
40
+ ```
41
+ pip install nemo_toolkit['all']
42
+ ```
43
+
44
+ ## How to Use this Model
45
+
46
+ The model is available for use in the NeMo toolkit [3], and can be used as a pre-trained checkpoint for inference or for fine-tuning on another dataset.
47
+
48
+ ### Automatically instantiate the model
49
+
50
+ ```python
51
+ import nemo.collections.asr as nemo_asr
52
+ asr_model = nemo_asr.models.ASRModel.from_pretrained("corygong/stt_en_conformer_ctc_small")
53
+ ```
54
+
55
+ ### Transcribing using Python
56
+ First, let's get a sample
57
+ ```
58
+ wget https://dldata-public.s3.us-east-2.amazonaws.com/2086-149220-0033.wav
59
+ ```
60
+ Then simply do:
61
+ ```
62
+ asr_model.transcribe(['2086-149220-0033.wav'])
63
+ ```
64
+
65
+ ### Transcribing many audio files
66
+
67
+ ```shell
68
+ python [NEMO_GIT_FOLDER]/examples/asr/transcribe_speech.py pretrained_name="corygong/stt_en_conformer_ctc_small" audio_dir="<DIRECTORY CONTAINING AUDIO FILES>"
69
+ ```
70
+
71
+ ### Input
72
+
73
+ This model accepts 16000 KHz Mono-channel Audio (wav files) as input.
74
+
75
+ ### Output
76
+
77
+ This model provides transcribed speech as a string for a given audio sample.
78
+
79
+ ## Model Architecture
80
+
81
+ <ADD SOME INFORMATION ABOUT THE ARCHITECTURE>
82
+
83
+ ## Training
84
+
85
+ <ADD INFORMATION ABOUT HOW THE MODEL WAS TRAINED - HOW MANY EPOCHS, AMOUNT OF COMPUTE ETC>
86
+
87
+ ### Datasets
88
+
89
+ <LIST THE NAME AND SPLITS OF DATASETS USED TO TRAIN THIS MODEL (ALONG WITH LANGUAGE AND ANY ADDITIONAL INFORMATION)>
90
+
91
+ ## Performance
92
+
93
+ <LIST THE SCORES OF THE MODEL -
94
+ OR
95
+ USE THE Hugging Face Evaluate LiBRARY TO UPLOAD METRICS>
96
+
97
+ ## Limitations
98
+
99
+ <DECLARE ANY POTENTIAL LIMITATIONS OF THE MODEL>
100
+
101
+ Eg:
102
+ Since this model was trained on publicly available speech datasets, the performance of this model might degrade for speech which includes technical terms, or vernacular that the model has not been trained on. The model might also perform worse for accented speech.
103
+
104
+
105
+ ## References
106
+
107
+ <ADD ANY REFERENCES HERE AS NEEDED>
108
+
109
+ [1] [NVIDIA NeMo Toolkit](https://github.com/NVIDIA/NeMo)
110
+