Update app.py

app.py

@@ -9,6 +9,11 @@ from modelscope.models.audio.tts import SambertHifigan
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 
+import torch
+import torchaudio
+from denoiser import pretrained
+from denoiser.dsp import convert_audio
+
 # model_0
 
 model_dir = os.path.abspath("./pretrain_work_dir")
@@ -108,6 +113,21 @@ def extend(audio):
 
     return "upsample.wav"
 
+# denoise
+
+model1 = pretrained.dns64().cuda()
+
+def denoise(audio):
+
+    wav, sr = torchaudio.load(audio)
+    wav = convert_audio(wav.cuda(), sr, model1.sample_rate, model1.chin)
+    with torch.no_grad():
+        denoised = model1(wav[None])[0]
+
+    write("denoised.wav", model1.sample_rate, denoised.data.cpu().numpy())
+    
+    return "denoised.wav"
+
 
 app = gr.Blocks()
 
@@ -125,11 +145,13 @@ with app:
             with gr.Row():
                 out = gr.Audio(label="为您生成的专属音频")
                 out1 = gr.Audio(label="更高采样率的专属音频")
+                out2 = gr.Audio(label="降噪后的高采样率音频")
             btn2 = gr.Button("一键提高采样率")
     
         btn.click(fn=infer, inputs=[inp], outputs=[out])
         btn1.click(fn=infer1, inputs=[inp], outputs=[out])
         btn2.click(fn=extend, inputs=[out], outputs=[out1])
+        btn2.click(fn=denoise, inputs=[out1], outputs=[out2])
 
     gr.Markdown("### <center>注意❗：请不要生成会对个人以及组织造成侵害的内容，此程序仅供科研、学习及个人娱乐使用。</center>")
     gr.HTML('''