app

app.py

@@ -60,7 +60,7 @@ def load_large_files(relative_path):
     #         file.write(chunk)
     #         bar.update(len(chunk))
     # print(f"File downloaded successfully and saved as: {file_path}")
-    file_path = hf_hub_download(repo_id="mingyuan/data_hf", filename=relative_path, repo_type="dataset", force_filename=relative_path.split('/')[-1])
+    file_path = hf_hub_download(repo_id="mingyuan/data_hf", filename=relative_path, repo_type="dataset")
     return file_path
 
 def motion_temporal_filter(motion, sigma=1):
@@ -105,10 +105,7 @@ if type == "local":
         "std": "../data_hf/data/motionverse/statistics/std.npy", 
         "skeleton": "../data_hf/data/motionverse/statistics/skeleton.npy", 
         "lmm": "../data_hf/data/motionverse/pretrained/lmm_small_demo.pth", 
-        "imagebind": "../data_hf/data/motionverse/pretrained/imagebind_huge.pth",
-        "audio_placeholder": "../data_hf/examples/placeholder.m4a",
-        "audio_surprise": "../data_hf/examples/surprise.m4a",
-        "audio_angry": "../data_hf/examples/angry.m4a"
+        "imagebind": "../data_hf/data/motionverse/pretrained/imagebind_huge.pth"
     }
     os.environ["NO_PROXY"] = os.environ["no_proxy"] = "localhost, 127.0.0.1:7860"
 else:
@@ -117,14 +114,15 @@ else:
         "std": "data/motionverse/statistics/std.npy", 
         "skeleton": "data/motionverse/statistics/skeleton.npy", 
         "lmm": "data/motionverse/pretrained/lmm_small_demo.pth", 
-        "imagebind": "data/motionverse/pretrained/imagebind_huge.pth",
-        "audio_placeholder": "examples/placeholder.m4a",
-        "audio_surprise": "examples/surprise.m4a",
-        "audio_angry": "examples/angry.m4a"
+        "imagebind": "data/motionverse/pretrained/imagebind_huge.pth"
     }
     load_file_list = {}
     for key in src_file_list.keys():
         load_file_list[key] = load_large_files(src_file_list[key])
+        
+load_file_list["audio_placeholder"] = "./examples/placeholder.m4a"
+load_file_list["audio_surprise"] = "./examples/surprise.m4a"
+load_file_list["audio_angry"] = "./examples/angry.m4a"
 
 model_lmm = create_lmm()
 model_imagebind = imagebind_huge(pretrained=True, ckpt_path=load_file_list["imagebind"])