Add application file

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+example/* filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,42 @@
+from transformers import pipeline
+import gradio as gr
+
+pipe = pipeline(
+    "audio-classification", model="juangtzi/wav2vec2-base-finetuned-gtzan"
+)
+
+def classify_audio(filepath):
+    import time
+    start_time = time.time()
+
+    preds = pipe(filepath)
+    outputs = {}
+    for p in preds:
+        outputs[p["label"]] = p["score"]
+
+    end_time = time.time()
+    prediction_time = end_time - start_time
+
+
+    return outputs, prediction_time
+
+
+title = "🎵 Music Genre Classifier"
+description = """
+Music Genre Classifier model (Fine-tuned "facebook/wav2vec2-base") Dataset: [GTZAN](https://huggingface.co/datasets/marsyas/gtzan)
+"""
+
+
+demo = gr.Interface(
+    fn=classify_audio,
+    inputs=gr.Audio(type="filepath"),
+    outputs=[gr.Label(), gr.Number(label="Prediction time (s)")],
+    title=title,
+    description=description,
+    examples="./example",
+    #cache_examples=True,
+    allow_flagging="never",
+)
+demo.queue()
+
+demo.launch(share=True)

example/example1.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a57f4a9a027afbc54e125bf2863c171163ece3db93fa2fc43b03c600d71ff837
+size 7356790

example/example2.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3bfdd222236b7e8f9013f72020f97327a22aea59a0a7884a4b33955e9840fdad
+size 7778915

example/example3.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5210e00887f2ae22633af46b3b33c1d298f4b35075822319560d45287ed821ef
+size 6798835

example/example4.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:374c3611af7a0de29d10a71f86c26feb1f18f4d4dab1bf62632184bdfc0dc148
+size 6645205

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+gradio
+torch