fix

app.py

@@ -71,11 +71,11 @@ def denoise(filename, ckpt_path = CHECKPOINT, out = "out.wav"):
     # inference
     noisy_audio = load_simple(filename)
 
-    for batch in tqdm(noisy_audio):
-        with torch.no_grad():
-            generated_audio = sampling(net, batch)
-            generated_audio = generated_audio.cpu()
-            sf.write(out, np.ravel(generated_audio.squeeze()), SAMPLE_RATE)
+    with torch.no_grad():
+        with torch.cuda.amp.autocast():
+            generated_audio = sampling(net, noisy_audio)
+            generated_audio = generated_audio[0].squeeze().cpu().numpy()
+            sf.write(out, np.ravel(generated_audio), SAMPLE_RATE)
 
     return out
 

requirements.txt

@@ -1,6 +1,6 @@
 pillow==6.2.0
 torch
-torchaudio==0.8.0
+torchaudio==0.13.1
 inflect==4.1.0
 soundfile
 scipy