Spaces:
Runtime error
Runtime error
Commit
Β·
1258aa9
1
Parent(s):
b1f349f
Update app.py (#12)
Browse files- Update app.py (3e6402fa2c6d394b367c1c0040af5b5c3f216c47)
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 |
-
|
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()
|