theodotus commited on
Commit
f083e13
·
1 Parent(s): c2163fe

Using squeezeformer

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,7 +8,7 @@ import nemo.collections.asr as nemo_asr
8
 
9
 
10
  asr_model = nemo_asr.models.EncDecCTCModelBPE. \
11
- from_pretrained("NeonBohdan/stt_uk_citrinet_512_gamma_0_25",map_location="cpu")
12
 
13
  asr_model.preprocessor.featurizer.dither = 0.0
14
  asr_model.preprocessor.featurizer.pad_to = 0
@@ -17,11 +17,11 @@ asr_model.encoder.freeze()
17
  asr_model.decoder.freeze()
18
 
19
 
20
- buffer_len = 1.6
21
  chunk_len = 0.8
22
  total_buffer = round(buffer_len * asr_model.cfg.sample_rate)
23
  overhead_len = round((buffer_len - chunk_len) * asr_model.cfg.sample_rate)
24
- model_stride = 8
25
 
26
 
27
 
 
8
 
9
 
10
  asr_model = nemo_asr.models.EncDecCTCModelBPE. \
11
+ from_pretrained("theodotus/stt_uk_squeezeformer_ctc_sm",map_location="cpu")
12
 
13
  asr_model.preprocessor.featurizer.dither = 0.0
14
  asr_model.preprocessor.featurizer.pad_to = 0
 
17
  asr_model.decoder.freeze()
18
 
19
 
20
+ buffer_len = 3.2
21
  chunk_len = 0.8
22
  total_buffer = round(buffer_len * asr_model.cfg.sample_rate)
23
  overhead_len = round((buffer_len - chunk_len) * asr_model.cfg.sample_rate)
24
+ model_stride = 4
25
 
26
 
27