Politrees commited on
Commit
1f30182
·
verified ·
1 Parent(s): 7da6b5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -21
app.py CHANGED
@@ -13,27 +13,27 @@ use_autocast = device == "cuda"
13
  # Roformer Models #
14
  #=========================#
15
  ROFORMER_MODELS = {
16
- 'BS Roformer | sdr 10.5309 by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
17
- 'BS Roformer | sdr 12.9628 by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
18
- 'BS Roformer | sdr 12.9755 by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
19
- 'BS Roformer | Chorus by Sucial': 'model_chorus_bs_roformer_ep_267_sdr_24.1275.ckpt',
20
- 'BS Roformer | DeReverb by anvuew': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
21
- 'Mel Roformer | sdr 11.4360 by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
22
- 'Mel Roformer | Vocals by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
23
- 'Mel Roformer | Crowd by Aufr33 and Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
24
- 'Mel Roformer | Karaoke by Aufr33 and Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
25
- 'Mel Roformer | DeNoise by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
26
- 'Mel Roformer | DeNoise Aggr by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.ckpt',
27
- 'MelBand Roformer | Aspiration by Sucial': 'aspiration_mel_band_roformer_sdr_18.9845.ckpt',
28
- 'MelBand Roformer | Aspiration Aggr by Sucial': 'aspiration_mel_band_roformer_less_aggr_sdr_18.1201.ckpt',
29
- 'MelBand Roformer | DeReverb by Anvuew': 'dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt',
30
- 'MelBand Roformer | DeReverb Aggr by Anvuew': 'dereverb_mel_band_roformer_less_aggressive_anvuew_sdr_18.8050.ckpt',
31
- 'MelBand Roformer | DeReverb-DeEcho v1 by Sucial': 'dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt',
32
- 'MelBand Roformer | DeReverb-DeEcho v2 by Sucial': 'dereverb_echo_mbr_v2_sdr_dry_13.4843.ckpt',
33
- 'MelBand Roformer Kim | Inst V1 by Unwa': 'melband_roformer_inst_v1.ckpt',
34
- 'MelBand Roformer Kim | Inst V1 (E) by Unwa': 'melband_roformer_inst_v1e.ckpt',
35
- 'MelBand Roformer Kim | Inst V2 by Unwa': 'melband_roformer_inst_v2.ckpt',
36
- 'MelBand Roformer Kim | InstVoc Duality by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
37
  }
38
  #=========================#
39
  # MDX23C Models #
 
13
  # Roformer Models #
14
  #=========================#
15
  ROFORMER_MODELS = {
16
+ 'BS Roformer | sdr 10.5309 - by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
17
+ 'BS Roformer | sdr 12.9628 - by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
18
+ 'BS Roformer | sdr 12.9755 - by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
19
+ 'BS Roformer | Chorus - by Sucial': 'model_chorus_bs_roformer_ep_267_sdr_24.1275.ckpt',
20
+ 'BS Roformer | DeReverb - by anvuew': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
21
+ 'Mel Roformer | sdr 11.4360 - by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
22
+ 'Mel Roformer | Vocals - by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
23
+ 'Mel Roformer | Crowd - by Aufr33 and Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
24
+ 'Mel Roformer | Karaoke - by Aufr33 and Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
25
+ 'Mel Roformer | DeNoise - by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
26
+ 'Mel Roformer | DeNoise Aggr - by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.ckpt',
27
+ 'MelBand Roformer | Aspiration - by Sucial': 'aspiration_mel_band_roformer_sdr_18.9845.ckpt',
28
+ 'MelBand Roformer | Aspiration Aggr - by Sucial': 'aspiration_mel_band_roformer_less_aggr_sdr_18.1201.ckpt',
29
+ 'MelBand Roformer | DeReverb - by Anvuew': 'dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt',
30
+ 'MelBand Roformer | DeReverb Aggr - by Anvuew': 'dereverb_mel_band_roformer_less_aggressive_anvuew_sdr_18.8050.ckpt',
31
+ 'MelBand Roformer | DeReverb-DeEcho - by Sucial': 'dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt',
32
+ 'MelBand Roformer Kim | Inst V1 - by Unwa': 'melband_roformer_inst_v1.ckpt',
33
+ 'MelBand Roformer Kim | Inst V1 (E) - by Unwa': 'melband_roformer_inst_v1e.ckpt',
34
+ 'MelBand Roformer Kim | Inst V2 - by Unwa': 'melband_roformer_inst_v2.ckpt',
35
+ 'MelBand Roformer Kim | InstVoc Duality V1 - by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
36
+ 'MelBand Roformer Kim | InstVoc Duality V2 - by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
37
  }
38
  #=========================#
39
  # MDX23C Models #