small change

app.py

@@ -136,7 +136,7 @@ def show_video(input_text, key_to_use):
                                                 gd_motion=2.0,
                                                 steps_num=300)
     edited_motion = diffout2motion(diff_out, normalizer).squeeze()
-    import ipdb; ipdb.set_trace()
+    # import ipdb; ipdb.set_trace()
     # aitrenderer = get_renderer()
     # SMPL_LAYER = SMPLLayer(model_type='smplh', ext='npz', gender='neutral')
     # edited_mot_to_render = pack_to_render(rots=edited_motion[..., 3:],
@@ -211,9 +211,10 @@ def download_motionfix():
 
 def download_motionfix_dataset():
     REPO_ID = 'athn-nik/example-model'
-    dataset_downloaded = hf_hub_download(REPO_ID, filename="tmed_compressed.ckpt")
-    import ipdb; ipdb.set_trace()
-    
+    dataset_downloaded_path = hf_hub_download(REPO_ID, filename="tmed_compressed.ckpt")
+    dataset_dict = joblib.load(dataset_downloaded_path)
+    return dataset_dict
+
 def download_embeddings():
     REPO_ID = 'athn-nik/example-model'
     # return hf_hub_download(REPO_ID, filename="min_checkpoint.ckpt")
@@ -224,7 +225,7 @@ def download_embeddings():
 
 MFIX_p = download_motionfix() + '/motionfix'
 SOURCE_MOTS_p = download_embeddings() + '/embeddings'
-MFIX_DATASET_p = download_motionfix_dataset() 
+MFIX_DATASET_DICT = download_motionfix_dataset() 
 
 import gradio as gr