File size: 619 Bytes
de7a838 252bac2 de7a838 252bac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from transformers import pipeline
import gradio as gr
model_id = "pollner/distilhubert-finetuned-ravdess"
classifier = pipeline("audio-classification", model=model_id)
def classify_audio(audio_file):
# audio_file is path or file object
result = classifier(audio_file)
return result
iface = gr.Interface(
fn=classify_audio,
inputs=gr.Audio(type="filepath"), # Removed 'source' parameter
outputs=gr.JSON(label="Classification result"),
title="Emotion recognition from speech (RAVDESS)",
description="Classifies emotion from audio using distilhubert-finetuned-ravdess"
)
iface.launch() |