Spaces:
Running
Running
Set beam size to 1
Browse files
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
|