Update app.py
Browse files
app.py
CHANGED
@@ -86,7 +86,7 @@ def handle_motion_generation(npy1,npy2):
|
|
86 |
|
87 |
config = get_config(config_path)
|
88 |
ae = get_autoencoder(config)
|
89 |
-
ae.load_state_dict(torch.load(checkpoint_path))
|
90 |
# ae.cuda()
|
91 |
ae.eval()
|
92 |
mean_pose, std_pose = get_meanpose("test", config.data)
|
|
|
86 |
|
87 |
config = get_config(config_path)
|
88 |
ae = get_autoencoder(config)
|
89 |
+
ae.load_state_dict(torch.load(checkpoint_path, map_location=torch.device('cpu')))
|
90 |
# ae.cuda()
|
91 |
ae.eval()
|
92 |
mean_pose, std_pose = get_meanpose("test", config.data)
|