Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,9 +28,9 @@ class TTS:
28
  self.flowgenerator.decoder.skip()
29
  self.flowgenerator.eval()
30
  if model_variant == '은식':
31
- last_chpt2 = './log/1038_eunsik_01/HiFI_GAN_00257000.pt'\
32
  elif model_variant == 'KSS':
33
- last_chpt1 = './log/KSS/HiFi_GAN_00135000.pt'
34
  check_point = torch.load(last_chpt2, map_location = device)
35
  self.voicegenerator.load_state_dict(check_point['gen_model'])
36
  self.voicegenerator.eval()
 
28
  self.flowgenerator.decoder.skip()
29
  self.flowgenerator.eval()
30
  if model_variant == '은식':
31
+ last_chpt2 = './log/1038_eunsik_01/HiFI_GAN_00257000.pt'
32
  elif model_variant == 'KSS':
33
+ last_chpt2 = './log/KSS/HiFi_GAN_00135000.pt'
34
  check_point = torch.load(last_chpt2, map_location = device)
35
  self.voicegenerator.load_state_dict(check_point['gen_model'])
36
  self.voicegenerator.eval()