Spaces:
Runtime error
Runtime error
File size: 440 Bytes
8d60679 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import gradio as gr
from transformers import pipeline
pipe = pipeline('audio-classification',model = "yojul/wav2vec2-base-one-shot-hip-hop-drums-clf")
def inference(file_path):
results = pipe(file_path)
return {x['label'] : x['score'] for x in results}
upl_audio = gr.Audio(sources="upload", type="filepath")
label = gr.Label(num_top_classes=5)
demo = gr.Interface(fn=inference, inputs=[upl_audio], outputs=[label])
demo.launch() |