fix

app.py

@@ -36,7 +36,7 @@ with open(CONFIG) as f:
     global trainset_config
     trainset_config         = config["trainset_config"]     # to read trainset configurations
 
-def denoise(filename, ckpt_path = CHECKPOINT):
+def denoise(filename, ckpt_path = CHECKPOINT, out = "out.wav"):
     """
     Denoise audio
     Parameters:
@@ -66,7 +66,6 @@ def denoise(filename, ckpt_path = CHECKPOINT):
 
     # inference
     batch_size = 1000000
-    new_file_name = "denoised_" + filename
     noisy_audio = load_simple(filename)
     LENGTH = len(noisy_audio[0].squeeze())
     noisy_audio = torch.chunk(noisy_audio, LENGTH // batch_size + 1, dim=1)
@@ -79,10 +78,9 @@ def denoise(filename, ckpt_path = CHECKPOINT):
             all_audio.append(generated_audio)
 
     all_audio = np.concatenate(all_audio, axis=0)
-    print("saved to:", new_file_name)
-    sf.write(new_file_name, np.ravel(all_audio.squeeze()), 32000)
+    sf.write(out, np.ravel(all_audio.squeeze()), 32000)
 
-    return new_file_name
+    return out
 
 
 audio = gr.inputs.Audio(label = "Audio to denoise", type = 'filepath')