mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
23 lines
529 B
Python
23 lines
529 B
Python
# Demo: (Audio) -> (Image)
|
|
|
|
import gradio as gr
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
from scipy import signal
|
|
|
|
|
|
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
|
|
|
|
|
|
iface = gr.Interface(spectrogram, "audio", "plot")
|
|
|
|
iface.test_launch()
|
|
if __name__ == "__main__":
|
|
iface.launch()
|