cymic commited on
Commit
7f6a9af
·
1 Parent(s): 411265d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -40,7 +40,7 @@ def get_text(text, hps):
40
  # hps_ms = utils.get_hparams_from_file("./configs/vctk_base.json")
41
 
42
 
43
- hps = utils.get_hparams_from_file("./configs/config.json")
44
  # net_g_ms = SynthesizerTrn(
45
  # len(symbols),
46
  # hps_ms.data.filter_length // 2 + 1,
@@ -49,7 +49,7 @@ hps = utils.get_hparams_from_file("./configs/config.json")
49
  # **hps_ms.model)
50
 
51
  net_g = SynthesizerTrn(
52
- len(hps.symbols),
53
  hps.data.filter_length // 2 + 1,
54
  hps.train.segment_size // hps.data.hop_length,
55
  **hps.model)
@@ -73,9 +73,8 @@ def tts_fn(text):
73
  stn_tst = get_text(text, hps)
74
  with torch.no_grad():
75
  x_tst = stn_tst.unsqueeze(0)
76
- x_tst_lengths = LongTensor([stn_tst.size(0)])
77
- audio = net_g.infer(x_tst, x_tst_lengths, noise_scale=.667, noise_scale_w=0.8, length_scale=1)[0][
78
- 0, 0].data.float().numpy()
79
  return "Success", (hps.data.sampling_rate, audio)
80
 
81
 
 
40
  # hps_ms = utils.get_hparams_from_file("./configs/vctk_base.json")
41
 
42
 
43
+ hps = utils.get_hparams_from_file("./configs/tokaiteio.json")
44
  # net_g_ms = SynthesizerTrn(
45
  # len(symbols),
46
  # hps_ms.data.filter_length // 2 + 1,
 
49
  # **hps_ms.model)
50
 
51
  net_g = SynthesizerTrn(
52
+ len(symbols),
53
  hps.data.filter_length // 2 + 1,
54
  hps.train.segment_size // hps.data.hop_length,
55
  **hps.model)
 
73
  stn_tst = get_text(text, hps)
74
  with torch.no_grad():
75
  x_tst = stn_tst.unsqueeze(0)
76
+ x_tst_lengths = torch.LongTensor([stn_tst.size(0)])
77
+ audio = net_g.infer(x_tst, x_tst_lengths, noise_scale=.667, noise_scale_w=0.8, length_scale=1)[0][0,0].data.float().numpy()
 
78
  return "Success", (hps.data.sampling_rate, audio)
79
 
80