Fixed missing model

vampnet/app.py

@@ -16,6 +16,33 @@ from vampnet.interface import Interface
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 os.chdir(SCRIPT_DIR)
 
+# 2. Define the models directory
+MODEL_DIR = SCRIPT_DIR / "models"
+MODEL_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def ensure_models_exist():
+    """Downloads weights from HF Hub if they aren't in the models/ folder."""
+    repo_id = "ProjectCETI/wham"
+    # List all the .pth files your app needs
+    files_to_download = ["codec.pth", "coarse.pth", "c2f.pth", "wavebeat.pth"]
+    print(f"Checking for model weights in {MODEL_DIR}...")
+    for filename in files_to_download:
+        target_file = MODEL_DIR / filename
+        if not target_file.exists():
+            print(f"Downloading {filename} from {repo_id}...")
+            hf_hub_download(
+                repo_id=repo_id,
+                filename=filename,
+                local_dir=str(MODEL_DIR),
+                local_dir_use_symlinks=False,
+            )
+        else:
+            print(f"✓ {filename} found.")
+
+
+ensure_models_exist()
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 sys.argv = ["app.py", "--args.load", "conf/interface.yml", "--Interface.device", device]