Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ from TTS.utils.manage import ModelManager
|
|
11 |
from TTS.tts.configs.xtts_config import XttsConfig
|
12 |
from TTS.tts.models.xtts import XttsAudioConfig
|
13 |
from TTS.config.shared_configs import BaseDatasetConfig
|
|
|
14 |
import torch.serialization
|
15 |
# torch.serialization.add_safe_globals([XttsConfig])
|
16 |
model_names = TTS().list_models()
|
@@ -21,7 +22,7 @@ model_name = "tts_models/multilingual/multi-dataset/xtts_v2" # move in v2, since
|
|
21 |
#m = ModelManager().download_model(model_name)
|
22 |
#print(m)
|
23 |
m = model_name
|
24 |
-
torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig])
|
25 |
tts = TTS(model_name, gpu=False)
|
26 |
tts.to("cpu") # no GPU or Amd
|
27 |
#tts.to("cuda") # cuda only
|
|
|
11 |
from TTS.tts.configs.xtts_config import XttsConfig
|
12 |
from TTS.tts.models.xtts import XttsAudioConfig
|
13 |
from TTS.config.shared_configs import BaseDatasetConfig
|
14 |
+
from TTS.tts.models.xtts import XttsArgs
|
15 |
import torch.serialization
|
16 |
# torch.serialization.add_safe_globals([XttsConfig])
|
17 |
model_names = TTS().list_models()
|
|
|
22 |
#m = ModelManager().download_model(model_name)
|
23 |
#print(m)
|
24 |
m = model_name
|
25 |
+
torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig, XttsArgs])
|
26 |
tts = TTS(model_name, gpu=False)
|
27 |
tts.to("cpu") # no GPU or Amd
|
28 |
#tts.to("cuda") # cuda only
|