Spaces:
Sleeping
Sleeping
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 ...')
|