mattricesound commited on
Commit
8151113
·
1 Parent(s): 69e7bce

Download ckpts with script

Browse files
Files changed (1) hide show
  1. app.py +1 -25
app.py CHANGED
@@ -22,32 +22,8 @@ def init_hydra():
22
 
23
 
24
  def download_checkpoints():
25
- os.mkdir("./ckpts")
26
  print("Downloading checkpoints")
27
- hydra.utils.download_url(
28
- "https://zenodo.org/record/8218621/files/classifier.ckpt?download=1",
29
- "./classifier.ckpt",
30
- )
31
- hydra.utils.download_url(
32
- "https://zenodo.org/record/8218621/files/dcunet_chorus_aug.ckpt?download=1",
33
- "./dcunet_chorus_aug.ckpt",
34
- )
35
- hydra.utils.download_url(
36
- "https://zenodo.org/record/8218621/files/dcunet_delay_aug.ckpt?download=1",
37
- "./dcunet_delay_aug.ckpt",
38
- )
39
- hydra.utils.download_url(
40
- "https://zenodo.org/record/8218621/files/dcunet_reverb_aug.ckpt?download=1",
41
- "./dcunet_reverb_aug.ckpt",
42
- )
43
- hydra.utils.download_url(
44
- "https://zenodo.org/record/8218621/files/demucs_compressor_aug.ckpt?download=1",
45
- "./demucs_compressor_aug.ckpt",
46
- )
47
- hydra.utils.download_url(
48
- "https://zenodo.org/record/8218621/files/demucs_distortion_aug.ckpt?download=1",
49
- "./demucs_distortion_aug.ckpt",
50
- )
51
 
52
 
53
  def load_models():
 
22
 
23
 
24
  def download_checkpoints():
 
25
  print("Downloading checkpoints")
26
+ os.system("scripts/download_ckpts.sh")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
 
28
 
29
  def load_models():