fix bug

app.py

@@ -38,7 +38,7 @@ with open(CONFIG) as f:
     global trainset_config
     trainset_config         = config["trainset_config"]     # to read trainset configurations
 
-def denoise(files, ckpt_path = CHECKPOINT):
+def denoise(filename, ckpt_path = CHECKPOINT):
     """
     Denoise audio
     Parameters:
@@ -68,25 +68,23 @@ def denoise(files, ckpt_path = CHECKPOINT):
 
     # inference
     batch_size = 1000000
-    for file_path in tqdm(files):
-        file_name = os.path.basename(file_path)
-        file_dir = os.path.dirname(file_name)
-        new_file_name = file_name + "_denoised.wav"
-        noisy_audio = load_simple(file_path)
-        LENGTH = len(noisy_audio[0].squeeze())
-        noisy_audio = torch.chunk(noisy_audio, LENGTH // batch_size + 1, dim=1)
-        all_audio = []
-
-        for batch in tqdm(noisy_audio):
-            with torch.no_grad():
-                generated_audio = sampling(net, batch)
-                generated_audio = generated_audio.cpu().numpy().squeeze()
-                all_audio.append(generated_audio)
-
-        all_audio = np.concatenate(all_audio, axis=0)
-        save_file = os.path.join(file_dir, new_file_name)
-        print("saved to:", save_file)
-        wavwrite(save_file, 32000, all_audio.squeeze())
+    new_file_name = filename + "_denoised.wav"
+    noisy_audio = load_simple(filename)
+    LENGTH = len(noisy_audio[0].squeeze())
+    noisy_audio = torch.chunk(noisy_audio, LENGTH // batch_size + 1, dim=1)
+    all_audio = []
+
+    for batch in tqdm(noisy_audio):
+        with torch.no_grad():
+            generated_audio = sampling(net, batch)
+            generated_audio = generated_audio.cpu().numpy().squeeze()
+            all_audio.append(generated_audio)
+
+    all_audio = np.concatenate(all_audio, axis=0)
+    print("saved to:", new_file_name)
+    wavwrite(new_file_name, 32000, all_audio.squeeze())
+
+    return new_file_name
 
 
 audio = gr.inputs.Audio(label = "Audio to denoise", type = 'filepath')