Fix spectrogram issue

app.py

@@ -113,7 +113,8 @@ def enhance(filepath, model_name):
         # enhanced
         D_clean = librosa.stft(out, n_fft=1024, hop_length=512)
         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=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)
         axs[1].set_title("Enhanced Spectrogram")
 
         plt.tight_layout()
@@ -148,5 +149,6 @@ with gr.Blocks() as demo:
         inputs=[input_audio, model_choice],
         outputs=[output_audio, plot_output]
     )
+    gr.Markdown("**Note**: The current models are trained on 16kHz audio. Therefore, any input audio not sampled at 16kHz will be automatically resampled before enhancement.")
 
 demo.queue().launch()