Download ckpts

app.py

@@ -4,6 +4,7 @@ import torchaudio
 import hydra
 from hydra import compose, initialize
 import random
+import os
 from remfx import effects
 
 cfg = None
@@ -20,6 +21,35 @@ def init_hydra():
     cfg = compose(config_name="config", overrides=["+exp=remfx_detect"])
 
 
+def download_checkpoints():
+    os.mkdir("./ckpts")
+    print("Downloading checkpoints")
+    hydra.utils.download_url(
+        "https://zenodo.org/record/8218621/files/classifier.ckpt?download=1",
+        "./classifier.ckpt",
+    )
+    hydra.utils.download_url(
+        "https://zenodo.org/record/8218621/files/dcunet_chorus_aug.ckpt?download=1",
+        "./dcunet_chorus_aug.ckpt",
+    )
+    hydra.utils.download_url(
+        "https://zenodo.org/record/8218621/files/dcunet_delay_aug.ckpt?download=1",
+        "./dcunet_delay_aug.ckpt",
+    )
+    hydra.utils.download_url(
+        "https://zenodo.org/record/8218621/files/dcunet_reverb_aug.ckpt?download=1",
+        "./dcunet_reverb_aug.ckpt",
+    )
+    hydra.utils.download_url(
+        "https://zenodo.org/record/8218621/files/demucs_compressor_aug.ckpt?download=1",
+        "./demucs_compressor_aug.ckpt",
+    )
+    hydra.utils.download_url(
+        "https://zenodo.org/record/8218621/files/demucs_distortion_aug.ckpt?download=1",
+        "./demucs_distortion_aug.ckpt",
+    )
+
+
 def load_models():
     global classifier
     print("Loading models")
@@ -177,5 +207,6 @@ def ui():
 
 if __name__ == "__main__":
     init_hydra()
+    download_checkpoints()
     load_models()
     ui()