w11wo commited on
Commit
9070b28
1 Parent(s): 66de3a6

added new model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ nltk.download("punkt")
15
  AUTH_TOKEN = os.environ["HF_TOKEN"]
16
 
17
  # download models
18
- url = hf_hub_url(repo_id="bookbot/grad-tts-en-ft-mixed", filename="grad_1000.pt")
19
  grad_tts_model_path = cached_download(url, use_auth_token=AUTH_TOKEN)
20
  torch.hub.download_url_to_file(
21
  "https://github.com/AK391/Speech-Backbones/releases/download/v1/hifigan.pt",
@@ -66,7 +66,7 @@ generator.load_state_dict(
66
  )
67
  _ = generator.eval()
68
 
69
- cmu = cmudict.CMUDict("./Grad-TTS/resources/cmu_dictionary_id_en")
70
 
71
  with open("./Grad-TTS/checkpts/hifigan-config.json") as f:
72
  h = AttrDict(json.load(f))
15
  AUTH_TOKEN = os.environ["HF_TOKEN"]
16
 
17
  # download models
18
+ url = hf_hub_url(repo_id="bookbot/grad-tts-en-ft-Weildan-v2", filename="grad_24000.pt")
19
  grad_tts_model_path = cached_download(url, use_auth_token=AUTH_TOKEN)
20
  torch.hub.download_url_to_file(
21
  "https://github.com/AK391/Speech-Backbones/releases/download/v1/hifigan.pt",
66
  )
67
  _ = generator.eval()
68
 
69
+ cmu = cmudict.CMUDict("./Grad-TTS/resources/cmu_dictionary_id")
70
 
71
  with open("./Grad-TTS/checkpts/hifigan-config.json") as f:
72
  h = AttrDict(json.load(f))