YinuoGuo27 commited on
Commit
fb797bd
·
verified ·
1 Parent(s): 3df8e39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -40,8 +40,7 @@ def predict(prompt, upload_reference_audio, microphone_reference_audio, referenc
40
  audio_file_pth = upload_reference_audio
41
  elif reference_audio_type == 'microphone':
42
  audio_file_pth = microphone_reference_audio
43
- torch.set_default_device('cuda')
44
- tts = TTS('tts_models/multilingual/multi-dataset/your_tts').to('cuda')
45
  tts.tts_to_file(
46
  text=prompt,
47
  file_path=output_file_path,
 
40
  audio_file_pth = upload_reference_audio
41
  elif reference_audio_type == 'microphone':
42
  audio_file_pth = microphone_reference_audio
43
+ tts = TTS('tts_models/multilingual/multi-dataset/your_tts')
 
44
  tts.tts_to_file(
45
  text=prompt,
46
  file_path=output_file_path,