Spaces:
Building
Building
update
Browse files
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 |
|