aka7774 commited on
Commit
9c7dd22
1 Parent(s): 26e9323

Update fn.py

Browse files
Files changed (1) hide show
  1. fn.py +2 -2
fn.py CHANGED
@@ -4,11 +4,11 @@ from audio_separator.separator import Separator
4
 
5
  separator = None
6
 
7
- def load_model():
8
  global separator
9
 
10
  separator = Separator()
11
- separator.load_model('Kim_Vocal_2.onnx')
12
 
13
  def run(audio_path):
14
  global separator
 
4
 
5
  separator = None
6
 
7
+ def load_model(model = 'MDX23C-8KFFT-InstVoc_HQ.ckpt'):
8
  global separator
9
 
10
  separator = Separator()
11
+ separator.load_model(model)
12
 
13
  def run(audio_path):
14
  global separator