Spaces:
Runtime error
Runtime error
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() |