Upload

app.py

@@ -25,6 +25,7 @@ import yaml
 import librosa
 import librosa.display
 import matplotlib
+import soundfile as sf
 from models.stfts    import mag_phase_stft, mag_phase_istft
 from models.generator import SEMamba
 from models.pcs400   import cal_pcs
@@ -80,7 +81,7 @@ def enhance(filepath):
             out = librosa.resample(out, orig_sr=16000, target_sr=orig_sr)
 
         # write file
-        sf.write("enhanced.wav", out, sr=orig_sr)
+        sf.write("enhanced.wav", out, orig_sr)
 
         # spectrograms
         fig, axs = plt.subplots(1, 2, figsize=(10, 4))