spectogram_main / run.py
aliabd's picture
aliabd HF Staff
Upload folder using huggingface_hub
5fbbfef verified
raw
history blame
492 Bytes
import matplotlib.pyplot as plt
import numpy as np
from scipy import signal
import gradio as gr
def spectrogram(audio):
sr, data = audio
if len(data.shape) == 2:
data = np.mean(data, axis=0)
frequencies, times, spectrogram_data = signal.spectrogram(
data, sr, window="hamming"
)
plt.pcolormesh(times, frequencies, np.log10(spectrogram_data))
return plt
demo = gr.Interface(spectrogram, "audio", "plot")
if __name__ == "__main__":
demo.launch()