Add auto device to automatically use GPU, add WebUI, update documentation, add link to live demo
This commit is contained in:
2
app.py
2
app.py
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user