csukuangfj commited on
Commit
386ef2a
·
1 Parent(s): 33085cc

fix examples

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  2. separate.py +1 -1
app.py CHANGED
@@ -32,14 +32,14 @@ import numpy as np
32
 
33
  from separate import get_file, load_audio, load_model, model_list
34
 
35
- examples = [
36
  "yesterday-once-more-Carpenters.mp3",
37
  "das-beste-Silbermond.mp3",
38
  "hotel-in-california.mp3",
39
  "起风了.mp3",
40
  ]
41
 
42
- for name in examples:
43
  filename = get_file(
44
  "csukuangfj/spleeter-torch",
45
  name,
@@ -48,6 +48,8 @@ for name in examples:
48
 
49
  shutil.copyfile(filename, name)
50
 
 
 
51
 
52
  def build_html_output(s: str, style: str = "result_item_success"):
53
  return f"""
 
32
 
33
  from separate import get_file, load_audio, load_model, model_list
34
 
35
+ wav_files = [
36
  "yesterday-once-more-Carpenters.mp3",
37
  "das-beste-Silbermond.mp3",
38
  "hotel-in-california.mp3",
39
  "起风了.mp3",
40
  ]
41
 
42
+ for name in wav_files:
43
  filename = get_file(
44
  "csukuangfj/spleeter-torch",
45
  name,
 
48
 
49
  shutil.copyfile(filename, name)
50
 
51
+ examples = [(model_list[0], w) for w in wav_files]
52
+
53
 
54
  def build_html_output(s: str, style: str = "result_item_success"):
55
  return f"""
separate.py CHANGED
@@ -120,9 +120,9 @@ def load_spleeter_model(name: str):
120
 
121
 
122
  model_list = [
 
123
  "sherpa-onnx-spleeter-2stems-fp16",
124
  "sherpa-onnx-spleeter-2stems-int8",
125
- "sherpa-onnx-spleeter-2stems",
126
  "UVR-MDX-NET-Inst_1.onnx",
127
  "UVR-MDX-NET-Inst_2.onnx",
128
  "UVR-MDX-NET-Inst_3.onnx",
 
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",