Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|