Deep50D commited on
Commit
1d3e6d1
·
verified ·
1 Parent(s): bf10f74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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