minor thing

app.py

@@ -27,25 +27,44 @@ except Exception as e:
 
 # Example videos
 example_videos = [
-    "./examples/000652_0_120.mp4",  # Replace with actual video paths
-    "./examples/000652_0_120.mp4",  # Replace with actual video paths
-    "./examples/000652_0_120.mp4",  # Replace with actual video paths
-    "./examples/000652_0_120.mp4",  # Replace with actual video paths
+    "./examples/001919.mp4",
+    "./examples/005376.mp4",
+    "./examples/001259.mp4",
+    "./examples/003686.mp4",
+    "./examples/001289.mp4",
+    "./examples/001893.mp4",
+    "./examples/003262.mp4",
+    "./examples/006117.mp4",
+    "./examples/001031.mp4",
+    "./examples/006247.mp4",
 ]
 # Example videos
 example_keys = [
-    "000091",  # Replace with actual video paths
-    "000091",  # Replace with actual video paths
-    "000091",  # Replace with actual video paths
-    "000091",  # Replace with actual video paths
+    "001919",
+    "005376",
+    "001259",
+    "003686",
+    "001289",
+    "001893",
+    "003262",
+    "006117",
+    "001031",
+    "006247",
 ]
 # Example videos
 example_texts = [
-    "need to use the opposite leg",  # Replace with actual video paths
-    "need to use the opposite leg2",  # Replace with actual video paths
-    "need to use the opposite leg3",  # Replace with actual video paths
-    "need to use the opposite leg4",  # Replace with actual video paths
+    "mirror",
+    "move in a smaller circle",
+    "less deep",
+    "turn back faster",
+    "cross your legs",
+    "step to the right",
+    "start sitting down a bit later",
+    "start a bit later, hold elbow lower at the end",
+    "extend the arm further back and catch higher",
+    "hold right arm higher",
 ]
+
 example_video_outputs = [gr.Video(label=f"Example {i+1}", 
                                   value=example_videos[i]) 
                          for i in range(4)]
@@ -55,9 +74,9 @@ class MotionEditor:
         # Don't initialize any CUDA components in __init__
         self.is_initialized = False
         self.MFIX_p = download_motionfix() + '/motionfix'
-        self.SOURCE_MOTS_p = download_embeddings() + '/embeddings'
+        # self.SOURCE_MOTS_p = download_embeddings() + '/embeddings'
         self.MFIX_DATASET_DICT = download_motionfix_dataset() 
-        self.model_ckpt_path = download_models()
+        self.model_ckpt_path = download_models("last_zipped")
         self.model_config_feats = download_model_config()
 
     @spaces.GPU

dataset_utils.py

@@ -1,8 +1,8 @@
 from gen_utils import read_json
 
 def load_motionfix(path_to_data):
-    dataset = read_json(path_to_data + '/amt_motionfix_latest.json')
-    splits = read_json(path_to_data + '/splits.json')
+    dataset = read_json(path_to_data + '/amt_motionfix_latest_final.json')
+    splits = read_json(path_to_data + '/splits_final.json')
 
     train_data = {}
     val_data = {}

download_deps.py

@@ -12,9 +12,9 @@ def get_smpl_models():
     return snapshot_download(repo_id=REPO_ID, allow_patterns="smplh*",
                              token=access_token_smpl)
 
-def download_models():
+def download_models(ckpt_to_dl):
     REPO_ID = 'athn-nik/example-model'
-    return hf_hub_download(REPO_ID, filename="tmed_compressed.ckpt")
+    return hf_hub_download(REPO_ID, filename=f"{ckpt_to_dl}.ckpt")
 
 def download_model_config():
     REPO_ID = 'athn-nik/example-model'