Add local checkpoint path and checkpoints folder

.gitattributes

@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 motion-diffusion-model/**/*.pt filter=lfs diff=lfs merge=lfs -text
 motion-diffusion-model/**/*.npy filter=lfs diff=lfs merge=lfs -text
 motion-diffusion-model/**/*.pkl filter=lfs diff=lfs merge=lfs -text
+checkpoints/*.pt filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,11 +1,9 @@
-cat > app.py << 'EOL'
 import gradio as gr
 import torch
 import os
 import sys
 import numpy as np
 import subprocess
-import gdown
 from pathlib import Path
 
 # Setup progress tracking
@@ -25,17 +23,6 @@ def setup_environment():
         progress_status = "Installing Spacy language model..."
         subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
     
-    # Create directories for checkpoints
-    os.makedirs("motion-diffusion-model/save/humanml_trans_enc_512", exist_ok=True)
-
-    # Download the model checkpoint if not already present
-    model_path = "motion-diffusion-model/save/humanml_trans_enc_512/model000200000.pt"
-    if not Path(model_path).exists():
-        progress_status = "Downloading model checkpoint (may take a few minutes)..."
-        # Model checkpoint from Google Drive
-        url = "https://drive.google.com/uc?id=1dbBIlDsYwvAcMctb3Zc_DLA-L-yrBNtN"
-        gdown.download(url, model_path, quiet=False)
-    
     # Download other necessary files if they don't exist
     if not Path("motion-diffusion-model/data/smpl").exists():
         progress_status = "Downloading SMPL files..."
@@ -82,8 +69,8 @@ def text_to_motion(text_prompt, motion_length=3.0, seed=0):
         if seed is not None:
             torch.manual_seed(seed)
         
-        # Model path
-        model_path = "motion-diffusion-model/save/humanml_trans_enc_512/model000200000.pt"
+        # Use the local checkpoint path
+        model_path = "checkpoints/mld_humanml.pt"
         
         # Generate the motion
         progress_status = "Running MDM generation..."
@@ -149,4 +136,4 @@ with gr.Blocks() as demo:
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()