Update open_oasis_master/generate.py

open_oasis_master/generate.py

@@ -39,8 +39,8 @@ ctx_max_noise_idx = ddim_noise_steps // 10 * 3
 # get input video 
 print(os.getcwd())
 video_id = "snippy-chartreuse-mastiff-f79998db196d-20220401-224517.chunk_001"
-mp4_path = f"{os.getcwd()}/sample_data/{video_id}.mp4"
-actions_path = f"{os.getcwd()}/sample_data/{video_id}.actions.pt"
+mp4_path = f"{os.getcwd()}/./sample_data/{video_id}.mp4"
+actions_path = f"{os.getcwd()}/./sample_data/{video_id}.actions.pt"
 video = read_video(mp4_path, pts_unit="sec")[0].float() / 255
 actions = one_hot_actions(torch.load(actions_path,map_location=torch.device('cpu')))
 offset = 100