Fix spectrogram issue

app.py

@@ -60,7 +60,8 @@ def enhance(filepath, model_name):
     model.eval()
     with torch.no_grad():
         # load & resample
-        wav, orig_sr = librosa.load(filepath, sr=None)
+        noisy_wav, orig_sr = librosa.load(filepath, sr=None)
+        wav = noisy_wav
         if orig_sr != 16000:
             wav = librosa.resample(wav, orig_sr=orig_sr, target_sr=16000)
         x = torch.from_numpy(wav).float().to(device)
@@ -102,7 +103,7 @@ def enhance(filepath, model_name):
         fig, axs = plt.subplots(1, 2, figsize=(16, 4))
 
         # noisy
-        D_noisy = librosa.stft(wav, n_fft=1024, hop_length=512)
+        D_noisy = librosa.stft(noisy_wav, n_fft=1024, hop_length=512)
         S_noisy = librosa.amplitude_to_db(np.abs(D_noisy), ref=np.max)
         librosa.display.specshow(S_noisy, sr=orig_sr, hop_length=512, x_axis="time", y_axis="hz", ax=axs[0], vmax=0)
         axs[0].set_title("Noisy Spectrogram")