Updating app file

app.py

@@ -61,13 +61,13 @@ def infer(model,data, notes):
     data= torch.tensor(data)
     if model == "CNN":
         model = MMCNN_CAT()
-        checkpoint = torch.load(MMCNN_CAT_ckpt_path)
+        checkpoint = torch.load(MMCNN_CAT_ckpt_path, map_location="cpu")
         model.load_state_dict(checkpoint['model_state_dict'])
         data = data.transpose(1,2).float()
 
     elif model == "RNN":
         model = MMRNN(device='cpu')
-        model.load_state_dict(torch.load(MMRNN_ckpt_path)['model_state_dict'])
+        model.load_state_dict(torch.load(MMRNN_ckpt_path, map_location="cpu")['model_state_dict'])
         data = data.float()
     model.eval()
     outputs, predicted = predict(model, data, embed_notes, device='cpu')