Hugging Face's logo

Spaces:
Politrees
/
audio-separator_UVR
Running

App Files Community
2
Politrees commited on
Commit
59e7cde
·
verified ·
1 Parent(s): dc76a20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +39 -26
app.py CHANGED
@@ -13,32 +13,43 @@ use_autocast = device == "cuda"
13
  # Roformer Models #
14
  #=========================#
15
  ROFORMER_MODELS = {
16
- #BS Roformer Models:
17
- 'BS Roformer | InstVoc v1 | by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
18
- 'BS Roformer | InstVoc v2 | by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
19
- 'BS Roformer | InstVoc v3 | by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
20
- 'BS Roformer | Chorus | by Sucial': 'model_chorus_bs_roformer_ep_267_sdr_24.1275.ckpt',
21
- 'BS Roformer | DeReverb | by anvuew': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
22
-
23
- #MelBand Roformer Models:
24
- 'MelBand Roformer | Inst V1 | by Unwa': 'melband_roformer_inst_v1.ckpt',
25
- 'MelBand Roformer | Inst V1 (E) | by Unwa': 'melband_roformer_inst_v1e.ckpt',
26
- 'MelBand Roformer | Inst V2 | by Unwa': 'melband_roformer_inst_v2.ckpt',
27
- 'MelBand Roformer | InstVoc | by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
28
- 'MelBand Roformer | InstVoc Duality V1 | by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
29
- 'MelBand Roformer | InstVoc Duality V2 | by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
30
- 'MelBand Roformer | Vocals | by KimberleyJensen': 'vocals_mel_band_roformer.ckpt',
31
- 'MelBand Roformer | Karaoke | by Aufr33 & Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
32
- 'MelBand Roformer | Crowd | by Aufr33 & Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
33
- 'MelBand Roformer | Aspiration | by Sucial': 'aspiration_mel_band_roformer_sdr_18.9845.ckpt',
34
- 'MelBand Roformer | Aspiration Aggr | by Sucial': 'aspiration_mel_band_roformer_less_aggr_sdr_18.1201.ckpt',
35
- 'MelBand Roformer | DeNoise | by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
36
- 'MelBand Roformer | DeNoise Aggr | by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.ckpt',
37
- 'MelBand Roformer | DeReverb | by Anvuew': 'dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt',
38
- 'MelBand Roformer | DeReverb Aggr | by Anvuew': 'dereverb_mel_band_roformer_less_aggressive_anvuew_sdr_18.8050.ckpt',
39
- 'MelBand Roformer | DeReverb-DeEcho | by Sucial': 'dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt',
 
 
 
 
 
 
 
 
 
 
 
 
40
  }
41
-
42
  #=========================#
43
  # MDX23C Models #
44
  #=========================#
@@ -46,6 +57,7 @@ MDX23C_MODELS = [
46
  'MDX23C-8KFFT-InstVoc_HQ.ckpt',
47
  'MDX23C-8KFFT-InstVoc_HQ_2.ckpt',
48
  'MDX23C_D1581.ckpt',
 
49
  ]
50
  #=========================#
51
  # MDXN-NET Models #
@@ -119,6 +131,7 @@ VR_ARCH_MODELS = [
119
  'UVR-DeEcho-DeReverb.pth',
120
  'UVR-De-Echo-Aggressive.pth',
121
  'UVR-De-Echo-Normal.pth',
 
122
  'UVR-DeNoise-Lite.pth',
123
  'UVR-DeNoise.pth',
124
  ]
@@ -368,7 +381,7 @@ with gr.Blocks(
368
  with gr.Tab("Roformer"):
369
  with gr.Group():
370
  with gr.Row():
371
- roformer_model = gr.Dropdown(label="Select the Model", choices=list(ROFORMER_MODELS.keys()))
372
  with gr.Row():
373
  roformer_seg_size = gr.Slider(minimum=32, maximum=4000, step=32, value=256, label="Segment Size", info="Larger consumes more resources, but may give better results.")
374
  roformer_override_seg_size = gr.Checkbox(value=False, label="Override segment size", info="Override model default segment size instead of using the model default value.")
 
13
  # Roformer Models #
14
  #=========================#
15
  ROFORMER_MODELS = {
16
+ # BS Roformer
17
+ 'BS Roformer | Chorus Male-Female 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 | sdr 10.5309 - by Viperx': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
20
+ 'BS Roformer | sdr 12.9628 - by Viperx': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
21
+ 'BS Roformer | sdr 12.9755 - by Viperx': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
22
+
23
+ # MelBand Roformer
24
+ 'MelBand Roformer | Aspiration by Sucial': 'aspiration_mel_band_roformer_sdr_18.9845.ckpt',
25
+ 'MelBand Roformer | Aspiration Less Aggressive by Sucial': 'aspiration_mel_band_roformer_less_aggr_sdr_18.1201.ckpt',
26
+ 'MelBand Roformer | Bleed Suppressor V1 by unwa-97chris': 'mel_band_roformer_bleed_suppressor_v1.ckpt',
27
+ 'MelBand Roformer | Crowd - by Aufr33 and Viperx': 'mel_band_roformer_crowd_aufr33_viperx_sdr_8.7144.ckpt',
28
+ 'MelBand Roformer | DeNoise - by Aufr33': 'denoise_mel_band_roformer_aufr33_sdr_27.9959.ckpt',
29
+ 'MelBand Roformer | DeNoise Aggr - by Aufr33': 'denoise_mel_band_roformer_aufr33_aggr_sdr_27.9768.ckpt',
30
+ 'MelBand Roformer | De-Reverb by anvuew': 'dereverb_mel_band_roformer_anvuew_sdr_19.1729.ckpt',
31
+ 'MelBand Roformer | De-Reverb Less Aggressive by anvuew': 'dereverb_mel_band_roformer_less_aggressive_anvuew_sdr_18.8050.ckpt',
32
+ 'MelBand Roformer | De-Reverb-Echo by Sucial': 'dereverb-echo_mel_band_roformer_sdr_10.0169.ckpt',
33
+ 'MelBand Roformer | De-Reverb-Echo V2 by Sucial': 'dereverb-echo_mel_band_roformer_sdr_13.4843_v2.ckpt',
34
+ 'MelBand Roformer | Karaoke - by Aufr33 and Viperx': 'mel_band_roformer_karaoke_aufr33_viperx_sdr_10.1956.ckpt',
35
+ 'MelBand Roformer | sdr 11.4360 - by Viperx': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt',
36
+ 'MelBand Roformer | Vocals by Kimberley Jensen': 'vocals_mel_band_roformer.ckpt',
37
+
38
+ # MelBand Roformer Kim
39
+ 'MelBand Roformer Kim | Big Beta 4 FT by unwa': 'melband_roformer_big_beta4.ckpt',
40
+ 'MelBand Roformer Kim | Big Beta 5e FT by unwa': 'melband_roformer_big_beta5e.ckpt',
41
+ 'MelBand Roformer Kim | Big SYHFT V1 by SYH99999': 'MelBandRoformerBigSYHFTV1.ckpt',
42
+ 'MelBand Roformer Kim | FT by unwa': 'mel_band_roformer_kim_ft_unwa.ckpt',
43
+ 'MelBand Roformer Kim | Inst V1 - by Unwa': 'melband_roformer_inst_v1.ckpt',
44
+ 'MelBand Roformer Kim | Inst V1 (E) - by Unwa': 'melband_roformer_inst_v1e.ckpt',
45
+ 'MelBand Roformer Kim | Inst V2 - by Unwa': 'melband_roformer_inst_v2.ckpt',
46
+ 'MelBand Roformer Kim | InstVoc Duality V1 - by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
47
+ 'MelBand Roformer Kim | InstVoc Duality V2 - by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
48
+ 'MelBand Roformer Kim | SYHFT V2 by SYH99999': 'MelBandRoformerSYHFTV2.ckpt',
49
+ 'MelBand Roformer Kim | SYHFT V2.5 by SYH99999': 'MelBandRoformerSYHFTV2.5.ckpt',
50
+ 'MelBand Roformer Kim | SYHFT V3 by SYH99999': 'MelBandRoformerSYHFTV3Epsilon.ckpt',
51
+ 'MelBand Roformer Kim | SYHFT by SYH99999': 'MelBandRoformerSYHFT.ckpt',
52
  }
 
53
  #=========================#
54
  # MDX23C Models #
55
  #=========================#
 
57
  'MDX23C-8KFFT-InstVoc_HQ.ckpt',
58
  'MDX23C-8KFFT-InstVoc_HQ_2.ckpt',
59
  'MDX23C_D1581.ckpt',
60
+ 'MDX23C-De-Reverb-aufr33-jarredou.ckpt',
61
  ]
62
  #=========================#
63
  # MDXN-NET Models #
 
131
  'UVR-DeEcho-DeReverb.pth',
132
  'UVR-De-Echo-Aggressive.pth',
133
  'UVR-De-Echo-Normal.pth',
134
+ 'UVR-De-Reverb-aufr33-jarredou.pth',
135
  'UVR-DeNoise-Lite.pth',
136
  'UVR-DeNoise.pth',
137
  ]
 
381
  with gr.Tab("Roformer"):
382
  with gr.Group():
383
  with gr.Row():
384
+ roformer_model = gr.Dropdown(value="MelBand Roformer Kim | Inst V1 (E) - by Unwa", label="Select the Model", choices=list(ROFORMER_MODELS.keys()))
385
  with gr.Row():
386
  roformer_seg_size = gr.Slider(minimum=32, maximum=4000, step=32, value=256, label="Segment Size", info="Larger consumes more resources, but may give better results.")
387
  roformer_override_seg_size = gr.Checkbox(value=False, label="Override segment size", info="Override model default segment size instead of using the model default value.")