Spaces:
Running
on
A10G
Running
on
A10G
Update scripts/download_models.py
Browse files
scripts/download_models.py
CHANGED
@@ -9,6 +9,9 @@ with open('config.yaml', 'r') as file:
|
|
9 |
ttm_model_size = config['AudioCraft']['ttm_model_size']
|
10 |
tta_model_size = config['AudioCraft']['tta_model_size']
|
11 |
|
|
|
|
|
|
|
12 |
|
13 |
# Downloading the TTS models
|
14 |
print('Step 1: Downloading TTS model ...')
|
|
|
9 |
ttm_model_size = config['AudioCraft']['ttm_model_size']
|
10 |
tta_model_size = config['AudioCraft']['tta_model_size']
|
11 |
|
12 |
+
# Download nltk
|
13 |
+
import nltk
|
14 |
+
nltk.download('punkt')
|
15 |
|
16 |
# Downloading the TTS models
|
17 |
print('Step 1: Downloading TTS model ...')
|