Spaces:
Running
Running
marigold334
commited on
Commit
•
b164712
1
Parent(s):
816e79a
Update app.py (#25)
Browse files- Update app.py (f74004c40fbe42437e224df89c516f7e5dcf222c)
app.py
CHANGED
@@ -27,7 +27,6 @@ class TTS:
|
|
27 |
last_chpt1 = './log/Taeyeon/Glow_TTS_400000.pt'
|
28 |
check_point = torch.load(last_chpt1, map_location = device)
|
29 |
self.flowgenerator.load_state_dict(check_point['generator' if model_variant != '태연' else 'model'])
|
30 |
-
self.flowgenerator.load_state_dict(check_point['generator'])
|
31 |
self.flowgenerator.decoder.skip()
|
32 |
self.flowgenerator.eval()
|
33 |
if model_variant == '감기걸린 은식':
|
|
|
27 |
last_chpt1 = './log/Taeyeon/Glow_TTS_400000.pt'
|
28 |
check_point = torch.load(last_chpt1, map_location = device)
|
29 |
self.flowgenerator.load_state_dict(check_point['generator' if model_variant != '태연' else 'model'])
|
|
|
30 |
self.flowgenerator.decoder.skip()
|
31 |
self.flowgenerator.eval()
|
32 |
if model_variant == '감기걸린 은식':
|