tjysdsg commited on
Commit
f5a907f
1 Parent(s): a202342

Set beam size to 1

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -16,6 +16,7 @@ VOCODER_TAG = 'espnet/cvss-c_en_wavegan_hubert_vocoder'
16
  VOCODER_DIR = 'vocoder'
17
 
18
  NGPU = 0
 
19
 
20
 
21
  def download_model(tag: str, out_dir: str):
@@ -81,6 +82,7 @@ def s2st(
81
  'config.yml',
82
  ),
83
  ngpu=NGPU,
 
84
  )
85
 
86
  # Restore working directory
 
16
  VOCODER_DIR = 'vocoder'
17
 
18
  NGPU = 0
19
+ BEAM_SIZE = 1
20
 
21
 
22
  def download_model(tag: str, out_dir: str):
 
82
  'config.yml',
83
  ),
84
  ngpu=NGPU,
85
+ beam_size=BEAM_SIZE,
86
  )
87
 
88
  # Restore working directory