Politrees commited on
Commit
96153f3
·
verified ·
1 Parent(s): 6c3efbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -21
app.py CHANGED
@@ -13,28 +13,32 @@ use_autocast = device == "cuda"
13
  # Roformer Models #
14
  #=========================#
15
  ROFORMER_MODELS = {
16
- 'BS Roformer | InstVoc v1 | by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
17
- 'BS Roformer | InstVoc v2 | by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
18
- 'BS Roformer | InstVoc v3 | 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
- 'MelBand Roformer | InstVoc | by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
22
- 'MelBand Roformer | Vocals | by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
23
- 'MelBand Roformer | Crowd | by Aufr33 & Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
24
- 'MelBand Roformer | Karaoke | by Aufr33 & Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
25
- 'MelBand Roformer | DeNoise | by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
26
- 'MelBand 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 | Inst V1 | by Unwa': 'melband_roformer_inst_v1.ckpt',
33
- 'MelBand Roformer | Inst V1 (E) | by Unwa': 'melband_roformer_inst_v1e.ckpt',
34
- 'MelBand Roformer | Inst V2 | by Unwa': 'melband_roformer_inst_v2.ckpt',
35
- 'MelBand Roformer | InstVoc Duality V1 | by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
36
- 'MelBand Roformer | InstVoc Duality V2 | by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
 
 
 
37
  }
 
38
  #=========================#
39
  # MDX23C Models #
40
  #=========================#
 
13
  # Roformer Models #
14
  #=========================#
15
  ROFORMER_MODELS = {
16
+ 'BS Roformer Models:': '',
17
+ 'BS Roformer | Chorus | by Sucial': 'model_chorus_bs_roformer_ep_267_sdr_24.1275.ckpt',
18
+ 'BS Roformer | DeReverb | by anvuew': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
19
+ 'BS Roformer | InstVoc v1 | by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
20
+ 'BS Roformer | InstVoc v2 | by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
21
+ 'BS Roformer | InstVoc v3 | by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
22
+ '': '',
23
+ 'MelBand Roformer Models:': '',
24
+ 'MelBand Roformer | Aspiration | by Sucial': 'aspiration_mel_band_roformer_sdr_18.9845.ckpt',
25
+ 'MelBand Roformer | Aspiration Aggr | by Sucial': 'aspiration_mel_band_roformer_less_aggr_sdr_18.1201.ckpt',
26
+ 'MelBand Roformer | Crowd | by Aufr33 & Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
27
+ 'MelBand Roformer | DeNoise | by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
28
+ 'MelBand Roformer | DeNoise Aggr | by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.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 | Inst V1 | by Unwa': 'melband_roformer_inst_v1.ckpt',
33
+ 'MelBand Roformer | Inst V1 (E) | by Unwa': 'melband_roformer_inst_v1e.ckpt',
34
+ 'MelBand Roformer | Inst V2 | by Unwa': 'melband_roformer_inst_v2.ckpt',
35
+ 'MelBand Roformer | InstVoc | by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
36
+ 'MelBand Roformer | InstVoc Duality V1 | by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
37
+ 'MelBand Roformer | InstVoc Duality V2 | by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
38
+ 'MelBand Roformer | Karaoke | by Aufr33 & Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
39
+ 'MelBand Roformer | Vocals | by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
40
  }
41
+
42
  #=========================#
43
  # MDX23C Models #
44
  #=========================#