kevinwang676 commited on
Commit
e7af428
·
1 Parent(s): d0ba148

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- #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')
 
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')