Update app.py

app.py

@@ -49,7 +49,7 @@ def handle_motion_generation(npy1,npy2):
 
     config = get_config(config_path)
     ae = get_autoencoder(config)
-    ae.load_state_dict(torch.load(checkpoint_path))
+    ae.load_state_dict(torch.load(checkpoint_path, map_location=torch.device('cpu')))
     # ae.cuda()
     ae.eval()
     mean_pose, std_pose = get_meanpose("test", config.data)