Spaces:
Running
Running
update
Browse files- examples/mpnet_aishell/run.sh +1 -1
- main.py +2 -2
examples/mpnet_aishell/run.sh
CHANGED
@@ -12,7 +12,7 @@ sh run.sh --stage 3 --stop_stage 3 --system_version centos --file_folder_name fi
|
|
12 |
--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
|
13 |
--speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
|
14 |
|
15 |
-
sh run.sh --stage
|
16 |
--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
|
17 |
--speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
|
18 |
|
|
|
12 |
--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
|
13 |
--speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
|
14 |
|
15 |
+
sh run.sh --stage 5 --stop_stage 5 --system_version centos --file_folder_name file_dir --final_model_name mpnet-aishell-20250224 \
|
16 |
--noise_dir "/data/tianxing/HuggingDatasets/nx_noise/data/noise" \
|
17 |
--speech_dir "/data/tianxing/HuggingDatasets/aishell/data_aishell/wav/train"
|
18 |
|
main.py
CHANGED
@@ -88,8 +88,8 @@ def main():
|
|
88 |
# engines
|
89 |
global denoise_engines
|
90 |
denoise_engines = {
|
91 |
-
"
|
92 |
-
pretrained_model_path_or_zip_file=(project_path / "trained_models/
|
93 |
),
|
94 |
}
|
95 |
|
|
|
88 |
# engines
|
89 |
global denoise_engines
|
90 |
denoise_engines = {
|
91 |
+
"mpnet_aishell_20250224": InferenceMPNet(
|
92 |
+
pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet_aishell_20250224.zip").as_posix(),
|
93 |
),
|
94 |
}
|
95 |
|