kevinwang676 commited on
Commit
d83143d
1 Parent(s): 641a3d4

Update app.py

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