Update denoisers/demucs.py

denoisers/demucs.py

@@ -85,6 +85,8 @@ class Demucs(torch.nn.Module):
         return x
 
     def predict(self, wav):
-        prediction = self.forward(torch.reshape(wav, (1, 1, -1)))
-        return prediction.detach()[0]
+        with torch.no_grad():
+            wav_reshaped = wav.reshape((1,1,-1))
+            prediction = self.forward(wav_reshaped)
+        return prediction[0]