Fix spectrogram issue

app.py

@@ -107,16 +107,16 @@ def enhance(filepath, model_name):
         fig, axs = plt.subplots(1, 2, figsize=(16, 4))
 
         # noisy
-        D_noisy = librosa.stft(noisy_wav, n_fft=1024, hop_length=512)
+        D_noisy = librosa.stft(noisy_wav, n_fft=512, hop_length=256)
         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)
+        librosa.display.specshow(S_noisy, sr=orig_sr, hop_length=256, x_axis="time", y_axis="hz", ax=axs[0], vmax=0)
         axs[0].set_title("Noisy Spectrogram")
 
         # enhanced
-        D_clean = librosa.stft(out_wav, n_fft=1024, hop_length=512)
+        D_clean = librosa.stft(out_wav, n_fft=512, hop_length=256)
         S_clean = librosa.amplitude_to_db(np.abs(D_clean), ref=np.max)
-        #librosa.display.specshow(S_clean, sr=orig_sr, hop_length=512, x_axis="time", y_axis="hz", ax=axs[1], vmax=0)
-        librosa.display.specshow(S_clean, sr=16000, hop_length=512, x_axis="time", y_axis="hz", ax=axs[1], vmax=0)
+        librosa.display.specshow(S_clean, sr=orig_sr, hop_length=256, x_axis="time", y_axis="hz", ax=axs[1], vmax=0)
+        #librosa.display.specshow(S_clean, sr=16000, hop_length=512, x_axis="time", y_axis="hz", ax=axs[1], vmax=0)
         axs[1].set_title("Enhanced Spectrogram")
 
         plt.tight_layout()