Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -55,17 +55,17 @@ ROFORMER_MODELS = {
|
|
55 |
#=========================#
|
56 |
# MDX23C Models #
|
57 |
#=========================#
|
58 |
-
MDX23C_MODELS =
|
59 |
'MDX23C-InstVoc HQ': 'MDX23C-8KFFT-InstVoc_HQ.ckpt',
|
60 |
'VIP | MDX23C-InstVoc HQ 2': 'MDX23C-8KFFT-InstVoc_HQ_2.ckpt',
|
61 |
'VIP | MDX23C_D1581': 'MDX23C_D1581.ckpt',
|
62 |
'MDX23C De-Reverb by aufr33-jarredou': 'MDX23C-De-Reverb-aufr33-jarredou.ckpt',
|
63 |
'MDX23C DrumSep by aufr33-jarredou': 'MDX23C-DrumSep-aufr33-jarredou.ckpt',
|
64 |
-
|
65 |
#=========================#
|
66 |
# MDXN-NET Models #
|
67 |
#=========================#
|
68 |
-
MDXNET_MODELS =
|
69 |
'UVR-MDX-NET Inst 1': 'UVR-MDX-NET-Inst_1.onnx',
|
70 |
'UVR-MDX-NET Inst 2': 'UVR-MDX-NET-Inst_2.onnx',
|
71 |
'UVR-MDX-NET Inst 3': 'UVR-MDX-NET-Inst_3.onnx',
|
@@ -105,11 +105,11 @@ MDXNET_MODELS = [
|
|
105 |
'Kim Vocal 1': 'Kim_Vocal_1.onnx',
|
106 |
'Kim Vocal 2': 'Kim_Vocal_2.onnx',
|
107 |
'Reverb HQ By FoxJoy': 'Reverb_HQ_By_FoxJoy.onnx',
|
108 |
-
|
109 |
#========================#
|
110 |
# VR-ARCH Models #
|
111 |
#========================#
|
112 |
-
VR_ARCH_MODELS =
|
113 |
'1_HP-UVR': '1_HP-UVR.pth',
|
114 |
'2_HP-UVR': '2_HP-UVR.pth',
|
115 |
'3_HP-Vocal-UVR': '3_HP-Vocal-UVR.pth',
|
@@ -138,16 +138,16 @@ VR_ARCH_MODELS = [
|
|
138 |
'MGM_MAIN_v4': 'MGM_MAIN_v4.pth',
|
139 |
'UVR-BVE-4B_SN-44100-1': 'UVR-BVE-4B_SN-44100-1.pth',
|
140 |
'UVR-De-Reverb by aufr33-jarredou': 'UVR-De-Reverb-aufr33-jarredou.pth',
|
141 |
-
|
142 |
#=======================#
|
143 |
# DEMUCS Models #
|
144 |
#=======================#
|
145 |
-
DEMUCS_MODELS =
|
146 |
'hdemucs_mmi': 'hdemucs_mmi.yaml',
|
147 |
'htdemucs': 'htdemucs.yaml',
|
148 |
'htdemucs_6s': 'htdemucs_6s.yaml',
|
149 |
'htdemucs_ft': 'htdemucs_ft.yaml',
|
150 |
-
|
151 |
|
152 |
def print_message(input_file, model_name):
|
153 |
"""Prints information about the audio separation process."""
|
|
|
55 |
#=========================#
|
56 |
# MDX23C Models #
|
57 |
#=========================#
|
58 |
+
MDX23C_MODELS = {
|
59 |
'MDX23C-InstVoc HQ': 'MDX23C-8KFFT-InstVoc_HQ.ckpt',
|
60 |
'VIP | MDX23C-InstVoc HQ 2': 'MDX23C-8KFFT-InstVoc_HQ_2.ckpt',
|
61 |
'VIP | MDX23C_D1581': 'MDX23C_D1581.ckpt',
|
62 |
'MDX23C De-Reverb by aufr33-jarredou': 'MDX23C-De-Reverb-aufr33-jarredou.ckpt',
|
63 |
'MDX23C DrumSep by aufr33-jarredou': 'MDX23C-DrumSep-aufr33-jarredou.ckpt',
|
64 |
+
}
|
65 |
#=========================#
|
66 |
# MDXN-NET Models #
|
67 |
#=========================#
|
68 |
+
MDXNET_MODELS = {
|
69 |
'UVR-MDX-NET Inst 1': 'UVR-MDX-NET-Inst_1.onnx',
|
70 |
'UVR-MDX-NET Inst 2': 'UVR-MDX-NET-Inst_2.onnx',
|
71 |
'UVR-MDX-NET Inst 3': 'UVR-MDX-NET-Inst_3.onnx',
|
|
|
105 |
'Kim Vocal 1': 'Kim_Vocal_1.onnx',
|
106 |
'Kim Vocal 2': 'Kim_Vocal_2.onnx',
|
107 |
'Reverb HQ By FoxJoy': 'Reverb_HQ_By_FoxJoy.onnx',
|
108 |
+
}
|
109 |
#========================#
|
110 |
# VR-ARCH Models #
|
111 |
#========================#
|
112 |
+
VR_ARCH_MODELS = {
|
113 |
'1_HP-UVR': '1_HP-UVR.pth',
|
114 |
'2_HP-UVR': '2_HP-UVR.pth',
|
115 |
'3_HP-Vocal-UVR': '3_HP-Vocal-UVR.pth',
|
|
|
138 |
'MGM_MAIN_v4': 'MGM_MAIN_v4.pth',
|
139 |
'UVR-BVE-4B_SN-44100-1': 'UVR-BVE-4B_SN-44100-1.pth',
|
140 |
'UVR-De-Reverb by aufr33-jarredou': 'UVR-De-Reverb-aufr33-jarredou.pth',
|
141 |
+
}
|
142 |
#=======================#
|
143 |
# DEMUCS Models #
|
144 |
#=======================#
|
145 |
+
DEMUCS_MODELS = {
|
146 |
'hdemucs_mmi': 'hdemucs_mmi.yaml',
|
147 |
'htdemucs': 'htdemucs.yaml',
|
148 |
'htdemucs_6s': 'htdemucs_6s.yaml',
|
149 |
'htdemucs_ft': 'htdemucs_ft.yaml',
|
150 |
+
}
|
151 |
|
152 |
def print_message(input_file, model_name):
|
153 |
"""Prints information about the audio separation process."""
|