Spaces:
Runtime error
Runtime error
Commit
·
d83143d
1
Parent(s):
641a3d4
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +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 |
-
device = 'cuda:0'
|
|
|
|
|
|
|
| 9 |
|
| 10 |
base_speaker_tts = BaseSpeakerTTS(f'{ckpt_base_en}/config.json', device=device)
|
| 11 |
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')
|