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."""
|