update

main.py

@@ -58,7 +58,20 @@ def shell(cmd: str):
     return Command.popen(cmd)
 
 
-denoise_engines = dict()
+denoise_engines = {
+    "mpnet-aishell-1-epoch": {
+        "infer_cls": InferenceMPNet,
+        "kwargs": {
+            "pretrained_model_path_or_zip_file": (project_path / "trained_models/mpnet-aishell-1-epoch.zip").as_posix()
+        }
+    },
+    "mpnet-aishell-11-epoch": {
+        "infer_cls": InferenceMPNet,
+        "kwargs": {
+            "pretrained_model_path_or_zip_file": (project_path / "trained_models/mpnet-aishell-11-epoch.zip").as_posix()
+        }
+    },
+}
 
 
 def when_click_denoise_button(noisy_audio_t, engine: str):
@@ -67,11 +80,15 @@ def when_click_denoise_button(noisy_audio_t, engine: str):
 
     noisy_audio = np.array(signal / (1 << 15), dtype=np.float32)
 
-    infer_engine = denoise_engines.get(engine)
-    if infer_engine is None:
+    infer_engine_param = denoise_engines.get(engine)
+    if infer_engine_param is None:
         raise gr.Error(f"invalid denoise engine: {engine}.")
 
     try:
+        infer_cls = infer_engine_param["infer_cls"]
+        kwargs = infer_engine_param["kwargs"]
+        infer_engine = infer_cls(**kwargs)
+
         enhanced_audio = infer_engine.enhancement_by_ndarray(noisy_audio)
         enhanced_audio = np.array(enhanced_audio * (1 << 15), dtype=np.int16)
     except Exception as e:
@@ -96,17 +113,6 @@ def main():
             token=args.hf_token,
         )
 
-    # engines
-    global denoise_engines
-    denoise_engines = {
-        "mpnet-aishell-1-epoch": InferenceMPNet(
-            pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet-aishell-1-epoch.zip").as_posix(),
-        ),
-        "mpnet-aishell-11-epoch": InferenceMPNet(
-            pretrained_model_path_or_zip_file=(project_path / "trained_models/mpnet-aishell-11-epoch.zip").as_posix(),
-        ),
-    }
-
     # choices
     denoise_engine_choices = list(denoise_engines.keys())
 
@@ -150,8 +156,8 @@ def main():
                     inputs=[dn_noisy_audio, dn_engine],
                     outputs=[dn_enhanced_audio],
                     fn=when_click_denoise_button,
-                    cache_examples=True,
-                    cache_mode="lazy",
+                    # cache_examples=True,
+                    # cache_mode="lazy",
                 )
 
             with gr.TabItem("shell"):