Add auto device to automatically use GPU, add WebUI, update documentation, add link to live demo

This commit is contained in:
mrfakename
2024-02-26 16:09:58 -08:00
committed by GitHub
parent 71d1249d16
commit d0fd6417c3
4 changed files with 25 additions and 8 deletions

2
app.py
View File

@@ -7,7 +7,7 @@ print("Make sure you've downloaded unidic (python -m unidic download) for this W
from melo.api import TTS
speed = 1.0
import tempfile
device = 'cuda' if torch.cuda.is_available() else 'cpu'
device = 'auto'
models = {
'EN': TTS(language='EN', device=device),
'ES': TTS(language='ES', device=device),