Politrees commited on
Commit
489e4d9
·
verified ·
1 Parent(s): 188b10f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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."""