Spaces:
Sleeping
Sleeping
kevinwang676
commited on
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')
|