HoneyTian commited on
Commit
5e7d9ca
·
1 Parent(s): 8ce0f99
Files changed (1) hide show
  1. main.py +10 -6
main.py CHANGED
@@ -46,12 +46,7 @@ def get_args():
46
  return args
47
 
48
 
49
- denoise_engines = {
50
- "mpnet": InferenceMPNet(
51
- pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet_aishell_20250221.zip").as_posix(),
52
- ),
53
-
54
- }
55
 
56
 
57
  def when_click_denoise_button(noisy_audio_t, engine: str):
@@ -88,6 +83,15 @@ def main():
88
  token=args.hf_token,
89
  )
90
 
 
 
 
 
 
 
 
 
 
91
  # choices
92
  denoise_engine_choices = list(denoise_engines.keys())
93
 
 
46
  return args
47
 
48
 
49
+ denoise_engines = dict()
 
 
 
 
 
50
 
51
 
52
  def when_click_denoise_button(noisy_audio_t, engine: str):
 
83
  token=args.hf_token,
84
  )
85
 
86
+ # engines
87
+ global denoise_engines
88
+ denoise_engines = {
89
+ "mpnet": InferenceMPNet(
90
+ pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet_aishell_20250221.zip").as_posix(),
91
+ ),
92
+
93
+ }
94
+
95
  # choices
96
  denoise_engine_choices = list(denoise_engines.keys())
97