Add download model

.gitignore

@@ -2,4 +2,6 @@ launchs/
 **/__pycache__
 sample/generated/
 .bash_history
-.config
+.config
+ckpt
+third_party

app.py

@@ -7,7 +7,11 @@ import os
 import sys
 import librosa
 import os.path as op
-PROJ_DIR = os.path.dirname(os.path.abspath(__file__))
+APP_DIR = op.dirname(op.abspath(__file__))
+
+from download import download_model
+# 下载模型
+download_model(APP_DIR)
 
 EXAMPLE_DESC = """female, dark, pop, sad, piano and drums, the bpm is 125."""
 EXAMPLE_LYRICS = """

download.py

@@ -0,0 +1,17 @@
+from huggingface_hub import snapshot_download
+import os
+
+def download_model(local_dir):
+    repo_id = "waytan22/SongGeneration-LeVo"
+
+    downloaded_path = snapshot_download(
+        repo_id=repo_id,
+        local_dir=local_dir,
+        revision="main",
+        token=os.environ.get("HF_TOKEN"), 
+        ignore_patterns=['.git*']
+    )
+    print(f"File downloaded to:{downloaded_path}")
+
+if __name__ == '__main__':
+    download_model('.')