Update app.py

app.py

@@ -18,13 +18,13 @@ def init_models(hparams):
     # load trained tacotron2 + GST model:
     model = load_model(hparams)
     checkpoint_path = "trained_models/checkpoint_78000.model"
-    model.load_state_dict(torch.load(checkpoint_path)['state_dict'])
+    model.load_state_dict(torch.load(checkpoint_path, map_location="cpu")['state_dict'])
     # model.to('cuda')
     _ = model.eval()
     
     # load pre trained MelGAN model for mel2audio:
     vocoder_checkpoint_path = "trained_models/nvidia_tacotron2_LJ11_epoch6400.pt"
-    checkpoint = torch.load(vocoder_checkpoint_path)
+    checkpoint = torch.load(vocoder_checkpoint_path, map_location="cpu")
     hp_melgan = load_hparam("melgan/config/default.yaml")
     vocoder_model = Generator(80)
     vocoder_model.load_state_dict(checkpoint['model_g'])