Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
pipe = pipeline( | |
"audio-classification", model="ceefax/distilhubert-finetuned-gtzan" | |
) | |
model_id = "ceefax/distilhubert-finetuned-gtzan" | |
pipe = pipeline("audio-classification", model=model_id) | |
title = "Genre Classifier" | |
description = "Genre classifier trained on GTZAN" | |
examples = ['Fleshy.mp3', 'Kombucha.mp3'] | |
interpretation='default' | |
enable_queue=True | |
def classify_audio(filepath): | |
preds = pipe(filepath) | |
outputs = {} | |
for p in preds: | |
outputs[p["label"]] = p["score"] | |
return outputs | |
demo = gr.Interface( | |
fn=classify_audio, inputs=gr.Audio(type="filepath"), outputs=gr.outputs.Label(), title=title,description=description | |
,examples=examples | |
,interpretation=interpretation,enable_queue=enable_queue | |
).launch() | |