Spaces:
Paused
Paused
Commit ·
e7af428
1
Parent(s): d0ba148
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,9 +5,9 @@ from api import BaseSpeakerTTS, ToneColorConverter
|
|
| 5 |
|
| 6 |
ckpt_base_en = 'checkpoints/checkpoints/base_speakers/EN'
|
| 7 |
ckpt_converter_en = 'checkpoints/checkpoints/converter'
|
| 8 |
-
|
| 9 |
|
| 10 |
-
device = "cpu"
|
| 11 |
|
| 12 |
base_speaker_tts = BaseSpeakerTTS(f'{ckpt_base_en}/config.json', device=device)
|
| 13 |
base_speaker_tts.load_ckpt(f'{ckpt_base_en}/checkpoint.pth')
|
|
|
|
| 5 |
|
| 6 |
ckpt_base_en = 'checkpoints/checkpoints/base_speakers/EN'
|
| 7 |
ckpt_converter_en = 'checkpoints/checkpoints/converter'
|
| 8 |
+
device = 'cuda:0'
|
| 9 |
|
| 10 |
+
#device = "cpu"
|
| 11 |
|
| 12 |
base_speaker_tts = BaseSpeakerTTS(f'{ckpt_base_en}/config.json', device=device)
|
| 13 |
base_speaker_tts.load_ckpt(f'{ckpt_base_en}/checkpoint.pth')
|