Hugging Face's logo

Spaces:
k2-fsa
/
source-separation
Running

App Files Community
csukuangfj commited on
Commit
d46145c
·
1 Parent(s): 6b7f648

add model info

Browse files
Files changed (1) hide show
  1. separate.py +23 -22
separate.py CHANGED
@@ -19,7 +19,7 @@ def convert_to_wav(in_filename: str) -> str:
19
 
20
  logging.info(f"Converting '{in_filename}' to '{out_filename}'")
21
  _ = os.system(
22
- f"ffmpeg -hide_banner -loglevel error -i '{in_filename}' -ar 441000 -ac 2 '{out_filename}' -y"
23
  )
24
 
25
  return out_filename
@@ -58,6 +58,7 @@ def get_file(
58
 
59
  @lru_cache(maxsize=30)
60
  def load_model(name: str):
 
61
  if "spleeter" in name:
62
  return load_spleeter_model(name)
63
  elif "UVR" in name:
@@ -120,25 +121,25 @@ def load_spleeter_model(name: str):
120
 
121
 
122
  model_list = [
123
- "sherpa-onnx-spleeter-2stems",
124
- "sherpa-onnx-spleeter-2stems-fp16",
125
- "sherpa-onnx-spleeter-2stems-int8",
126
- "UVR-MDX-NET-Inst_1.onnx",
127
- "UVR-MDX-NET-Inst_2.onnx",
128
- "UVR-MDX-NET-Inst_3.onnx",
129
- "UVR-MDX-NET-Inst_HQ_1.onnx",
130
- "UVR-MDX-NET-Inst_HQ_2.onnx",
131
- "UVR-MDX-NET-Inst_HQ_3.onnx",
132
- "UVR-MDX-NET-Inst_HQ_4.onnx",
133
- "UVR-MDX-NET-Inst_HQ_5.onnx",
134
- "UVR-MDX-NET-Inst_Main.onnx",
135
- "UVR-MDX-NET-Voc_FT.onnx",
136
- "UVR-MDX-NET_Crowd_HQ_1.onnx",
137
- "UVR_MDXNET_1_9703.onnx",
138
- "UVR_MDXNET_2_9682.onnx",
139
- "UVR_MDXNET_3_9662.onnx",
140
- "UVR_MDXNET_9482.onnx",
141
- "UVR_MDXNET_KARA.onnx",
142
- "UVR_MDXNET_KARA_2.onnx",
143
- "UVR_MDXNET_Main.onnx",
144
  ]
 
19
 
20
  logging.info(f"Converting '{in_filename}' to '{out_filename}'")
21
  _ = os.system(
22
+ f"ffmpeg -hide_banner -loglevel error -i '{in_filename}' -ar 44100 -ac 2 '{out_filename}' -y"
23
  )
24
 
25
  return out_filename
 
58
 
59
  @lru_cache(maxsize=30)
60
  def load_model(name: str):
61
+ name = name.split("|")[0]
62
  if "spleeter" in name:
63
  return load_spleeter_model(name)
64
  elif "UVR" in name:
 
121
 
122
 
123
  model_list = [
124
+ "sherpa-onnx-spleeter-2stems|fastest",
125
+ "sherpa-onnx-spleeter-2stems-fp16|fastest",
126
+ "sherpa-onnx-spleeter-2stems-int8|fastest",
127
+ "UVR_MDXNET_1_9703.onnx|slow",
128
+ "UVR_MDXNET_2_9682.onnx|slow",
129
+ "UVR_MDXNET_3_9662.onnx|slow",
130
+ "UVR_MDXNET_9482.onnx|slow",
131
+ "UVR_MDXNET_KARA.onnx|slow",
132
+ "UVR_MDXNET_KARA_2.onnx|slowest",
133
+ "UVR_MDXNET_Main.onnx|slowest",
134
+ "UVR-MDX-NET-Inst_1.onnx|slowest",
135
+ "UVR-MDX-NET-Inst_2.onnx|slowest",
136
+ "UVR-MDX-NET-Inst_3.onnx|slowest",
137
+ "UVR-MDX-NET-Inst_HQ_1.onnx|slowest",
138
+ "UVR-MDX-NET-Inst_HQ_2.onnx|slowest",
139
+ "UVR-MDX-NET-Inst_HQ_3.onnx|slowest",
140
+ "UVR-MDX-NET-Inst_HQ_4.onnx|slowest",
141
+ "UVR-MDX-NET-Inst_HQ_5.onnx|slowest",
142
+ "UVR-MDX-NET-Inst_Main.onnx|slowest",
143
+ "UVR-MDX-NET-Voc_FT.onnx|slowest",
144
+ "UVR-MDX-NET_Crowd_HQ_1.onnx|slowest",
145
  ]