Update app.py

app.py

@@ -50,7 +50,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.cuda()
+    # ae.cuda()
     ae.eval()
     mean_pose, std_pose = get_meanpose("test", config.data)
     # print("loaded model")
@@ -60,16 +60,18 @@ def handle_motion_generation(npy1,npy2):
 
     os.makedirs(out_dir_path, exist_ok=True)
 
-    # path1 = '/home/fazhong/studio/transmomo.pytorch/data/mixamo/36_800_24/test/PUMPKINHULK_L/Back_Squat/motions/2.npy'
-    # path2 = '/home/fazhong/studio/transmomo.pytorch/data/mixamo/36_800_24/test/PUMPKINHULK_L/Golf_Post_Shot/motions/3.npy'
+    path1 = './data/1.npy'
+    path2 = './data/2.npy'
     out_path1 = os.path.join(out_dir_path, "adv.npy")
 
 
     x_a, x_a_start = load_and_preprocess(path1, config, mean_pose, std_pose)
     x_b, x_b_start = load_and_preprocess(path2, config, mean_pose, std_pose)
     
-    x_a_batch = x_a.unsqueeze(0).cuda()
-    x_b_batch = x_b.unsqueeze(0).cuda()
+    # x_a_batch = x_a.unsqueeze(0).cuda()
+    # x_b_batch = x_b.unsqueeze(0).cuda()
+    x_a_batch = x_a.unsqueeze(0)
+    x_b_batch = x_b.unsqueeze(0)
 
     x_ab = ae.cross2d(x_a_batch, x_b_batch, x_a_batch)
     x_ab = postprocess(x_ab, mean_pose, std_pose, config.data.unit, start=x_a_start)
@@ -88,7 +90,6 @@ def handle_motion_generation(npy1,npy2):
     motion2video(motion_data, height, width, save_path, colors, bg_color=bg_color, transparency=False, fps=fps)
     first_frame_image = Image.open('./an-frames/0000.png')
     return first_frame_image
-    # print('hi')
 
 with gr.Blocks() as demo:
     gr.Markdown("Upload two `.npy` files to generate motion and visualize the first frame of the output animation.")