Spaces:
Sleeping
Sleeping
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
|