YinuoGuo27 commited on
Commit
4b67b88
·
verified ·
1 Parent(s): 26ece02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -18,6 +18,8 @@ for root, dirs, files in os.walk(local_dir):
18
  file_path = os.path.join(root, file)
19
  print(file_path)
20
 
 
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  torch.set_default_device(device)
23
  tts = TTS('tts_models/multilingual/multi-dataset/your_tts').to(device)
 
18
  file_path = os.path.join(root, file)
19
  print(file_path)
20
 
21
+ print('torch.cuda.is_available()', torch.cuda.is_available())
22
+
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
24
  torch.set_default_device(device)
25
  tts = TTS('tts_models/multilingual/multi-dataset/your_tts').to(device)