Update app.py

app.py

@@ -21,15 +21,15 @@ class TTS:
         if model_variant == '은식':
             name = '1038_eunsik_01'
             last_chpt1 = './log/1038_eunsik_01/Glow_TTS_00289602.pt'
-        check_point = torch.load(last_chpt1)
-        self.flowgenerator.load_state_dict(check_point['generator'], map_location = device)
+        check_point = torch.load(last_chpt1, map_location = device)
+        self.flowgenerator.load_state_dict(check_point['generator'])
         self.flowgenerator.decoder.skip()
         self.flowgenerator.eval()
         if model_variant == '은식':
             name = '1038_eunsik_01'
             last_chpt2 = './log/1038_eunsik_01/HiFI_GAN_00257000.pt'
-        check_point = torch.load(last_chpt2)
-        self.voicegenerator.load_state_dict(check_point['gen_model'], map_location = device)
+        check_point = torch.load(last_chpt2, map_location = device)
+        self.voicegenerator.load_state_dict(check_point['gen_model'])
         self.voicegenerator.eval()
         self.voicegenerator.remove_weight_norm()