marigold334 commited on
Commit
64e6fa3
β€’
1 Parent(s): ecc0d3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ class TTS:
25
  elif model_variant == 'KSS':
26
  last_chpt1 = './log/KSS/Glow_TTS_00280641.pt'
27
  elif model_variant == 'νƒœμ—°':
28
- last_chpt1 = './log/Taeyeon/Glow_TTS_400000.pt'
29
  check_point = torch.load(last_chpt1, map_location = device)
30
  self.flowgenerator.load_state_dict(check_point['generator' if model_variant != 'νƒœμ—°' else 'model'])
31
  self.flowgenerator.decoder.skip()
@@ -35,7 +35,7 @@ class TTS:
35
  elif model_variant == 'KSS':
36
  last_chpt2 = './log/KSS/HiFi_GAN_00135000.pt'
37
  elif model_variant == 'νƒœμ—°':
38
- last_chpt2 = './log/Taeyeon/HiFi_GAN_337000.pt'
39
  check_point = torch.load(last_chpt2, map_location = device)
40
  self.voicegenerator.load_state_dict(check_point['gen_model'])
41
  self.voicegenerator.eval()
 
25
  elif model_variant == 'KSS':
26
  last_chpt1 = './log/KSS/Glow_TTS_00280641.pt'
27
  elif model_variant == 'νƒœμ—°':
28
+ last_chpt1 = './log/Taeyeon/Glow_TTS_337000.pt'
29
  check_point = torch.load(last_chpt1, map_location = device)
30
  self.flowgenerator.load_state_dict(check_point['generator' if model_variant != 'νƒœμ—°' else 'model'])
31
  self.flowgenerator.decoder.skip()
 
35
  elif model_variant == 'KSS':
36
  last_chpt2 = './log/KSS/HiFi_GAN_00135000.pt'
37
  elif model_variant == 'νƒœμ—°':
38
+ last_chpt2 = './log/Taeyeon/HiFi_GAN_400000.pt'
39
  check_point = torch.load(last_chpt2, map_location = device)
40
  self.voicegenerator.load_state_dict(check_point['gen_model'])
41
  self.voicegenerator.eval()