Upload with huggingface_hub

README.md

@@ -6,6 +6,6 @@ colorFrom: indigo
 colorTo: indigo
 sdk: gradio
 sdk_version: 3.4.1
-app_file: app.py
+app_file: run.py
 pinned: false
 ---

run.py

@@ -0,0 +1,56 @@
+from math import log2, pow
+import os
+
+import numpy as np
+from scipy.fftpack import fft
+
+import gradio as gr
+
+A4 = 440
+C0 = A4 * pow(2, -4.75)
+name = ["C", "C#", "D", "D#", "E", "F", "F#", "G", "G#", "A", "A#", "B"]
+
+
+def get_pitch(freq):
+    h = round(12 * log2(freq / C0))
+    n = h % 12
+    return name[n]
+
+
+def main_note(audio):
+    rate, y = audio
+    if len(y.shape) == 2:
+        y = y.T[0]
+    N = len(y)
+    T = 1.0 / rate
+    x = np.linspace(0.0, N * T, N)
+    yf = fft(y)
+    yf2 = 2.0 / N * np.abs(yf[0 : N // 2])
+    xf = np.linspace(0.0, 1.0 / (2.0 * T), N // 2)
+
+    volume_per_pitch = {}
+    total_volume = np.sum(yf2)
+    for freq, volume in zip(xf, yf2):
+        if freq == 0:
+            continue
+        pitch = get_pitch(freq)
+        if pitch not in volume_per_pitch:
+            volume_per_pitch[pitch] = 0
+        volume_per_pitch[pitch] += 1.0 * volume / total_volume
+    volume_per_pitch = {k: float(v) for k, v in volume_per_pitch.items()}
+    return volume_per_pitch
+
+
+demo = gr.Interface(
+    main_note,
+    gr.Audio(source="microphone"),
+    gr.Label(num_top_classes=4),
+    examples=[
+        [os.path.join(os.path.dirname(__file__),"audio/recording1.wav")],
+        [os.path.join(os.path.dirname(__file__),"audio/cantina.wav")],
+    ],
+    interpretation="default",
+)
+
+if __name__ == "__main__":
+    demo.launch()