Update app.py

app.py

@@ -22,14 +22,14 @@ class TTS:
             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'])
+        self.flowgenerator.load_state_dict(check_point['generator'], map_location = device)
         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'])
+        self.voicegenerator.load_state_dict(check_point['gen_model'], map_location = device)
         self.voicegenerator.eval()
         self.voicegenerator.remove_weight_norm()