Hugging Face's logo

Spaces:
netynet
/
UVR-UI
Sleeping

App Files Community
Next commited on
Commit
dfdab28
·
verified ·
1 Parent(s): f9747a1

Create uvrmodel.py

Browse files
Files changed (1) hide show
  1. uvrmodel.py +115 -0
uvrmodel.py ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ roformer_models = {
2
+ 'BS-Roformer-Viperx-1297.ckpt': 'model_bs_roformer_ep_317_sdr_12.9755.ckpt',
3
+ 'BS-Roformer-Viperx-1296.ckpt': 'model_bs_roformer_ep_368_sdr_12.9628.ckpt',
4
+ 'BS-Roformer-Viperx-1053.ckpt': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
5
+ 'Mel-Roformer-Viperx-1143.ckpt': 'model_mel_band_roformer_ep_3005_sdr_11.4360.ckpt'
6
+ }
7
+
8
+ mdx23c_models = [
9
+ 'MDX23C_D1581.ckpt',
10
+ 'MDX23C-8KFFT-InstVoc_HQ.ckpt',
11
+ 'MDX23C-8KFFT-InstVoc_HQ_2.ckpt',
12
+ ]
13
+
14
+ mdxnet_models = [
15
+ 'UVR-MDX-NET-Inst_full_292.onnx',
16
+ 'UVR-MDX-NET_Inst_187_beta.onnx',
17
+ 'UVR-MDX-NET_Inst_82_beta.onnx',
18
+ 'UVR-MDX-NET_Inst_90_beta.onnx',
19
+ 'UVR-MDX-NET_Main_340.onnx',
20
+ 'UVR-MDX-NET_Main_390.onnx',
21
+ 'UVR-MDX-NET_Main_406.onnx',
22
+ 'UVR-MDX-NET_Main_427.onnx',
23
+ 'UVR-MDX-NET_Main_438.onnx',
24
+ 'UVR-MDX-NET-Inst_HQ_1.onnx',
25
+ 'UVR-MDX-NET-Inst_HQ_2.onnx',
26
+ 'UVR-MDX-NET-Inst_HQ_3.onnx',
27
+ 'UVR-MDX-NET-Inst_HQ_4.onnx',
28
+ 'UVR_MDXNET_Main.onnx',
29
+ 'UVR-MDX-NET-Inst_Main.onnx',
30
+ 'UVR_MDXNET_1_9703.onnx',
31
+ 'UVR_MDXNET_2_9682.onnx',
32
+ 'UVR_MDXNET_3_9662.onnx',
33
+ 'UVR-MDX-NET-Inst_1.onnx',
34
+ 'UVR-MDX-NET-Inst_2.onnx',
35
+ 'UVR-MDX-NET-Inst_3.onnx',
36
+ 'UVR_MDXNET_KARA.onnx',
37
+ 'UVR_MDXNET_KARA_2.onnx',
38
+ 'UVR_MDXNET_9482.onnx',
39
+ 'UVR-MDX-NET-Voc_FT.onnx',
40
+ 'Kim_Vocal_1.onnx',
41
+ 'Kim_Vocal_2.onnx',
42
+ 'Kim_Inst.onnx',
43
+ 'Reverb_HQ_By_FoxJoy.onnx',
44
+ 'UVR-MDX-NET_Crowd_HQ_1.onnx',
45
+ 'kuielab_a_vocals.onnx',
46
+ 'kuielab_a_other.onnx',
47
+ 'kuielab_a_bass.onnx',
48
+ 'kuielab_a_drums.onnx',
49
+ 'kuielab_b_vocals.onnx',
50
+ 'kuielab_b_other.onnx',
51
+ 'kuielab_b_bass.onnx',
52
+ 'kuielab_b_drums.onnx',
53
+ ]
54
+
55
+ vrarch_models = [
56
+ '1_HP-UVR.pth',
57
+ '2_HP-UVR.pth',
58
+ '3_HP-Vocal-UVR.pth',
59
+ '4_HP-Vocal-UVR.pth',
60
+ '5_HP-Karaoke-UVR.pth',
61
+ '6_HP-Karaoke-UVR.pth',
62
+ '7_HP2-UVR.pth',
63
+ '8_HP2-UVR.pth',
64
+ '9_HP2-UVR.pth',
65
+ '10_SP-UVR-2B-32000-1.pth',
66
+ '11_SP-UVR-2B-32000-2.pth',
67
+ '12_SP-UVR-3B-44100.pth',
68
+ '13_SP-UVR-4B-44100-1.pth',
69
+ '14_SP-UVR-4B-44100-2.pth',
70
+ '15_SP-UVR-MID-44100-1.pth',
71
+ '16_SP-UVR-MID-44100-2.pth',
72
+ '17_HP-Wind_Inst-UVR.pth',
73
+ 'UVR-De-Echo-Aggressive.pth',
74
+ 'UVR-De-Echo-Normal.pth',
75
+ 'UVR-DeEcho-DeReverb.pth',
76
+ 'UVR-DeNoise-Lite.pth',
77
+ 'UVR-DeNoise.pth',
78
+ 'UVR-BVE-4B_SN-44100-1.pth',
79
+ 'MGM_HIGHEND_v4.pth',
80
+ 'MGM_LOWEND_A_v4.pth',
81
+ 'MGM_LOWEND_B_v4.pth',
82
+ 'MGM_MAIN_v4.pth',
83
+ ]
84
+
85
+ demucs_models = [
86
+ 'htdemucs_ft.yaml',
87
+ 'htdemucs.yaml',
88
+ 'hdemucs_mmi.yaml',
89
+ ]
90
+
91
+ output_format = [
92
+ 'wav',
93
+ 'flac',
94
+ 'mp3',
95
+ ]
96
+
97
+ mdxnet_overlap_values = [
98
+ '0.25',
99
+ '0.5',
100
+ '0.75',
101
+ '0.99',
102
+ ]
103
+
104
+ vrarch_window_size_values = [
105
+ '320',
106
+ '512',
107
+ '1024',
108
+ ]
109
+
110
+ demucs_overlap_values = [
111
+ '0.25',
112
+ '0.50',
113
+ '0.75',
114
+ '0.99',
115
+ ]