import gradio as gr | |
from transformers import pipeline | |
pipe = pipeline(task="audio-classification", model="thangtrungnguyen/vietnamese-female-male-voice-classification-model") | |
def classify_audio(filepath): | |
preds = pipe(filepath) | |
outputs = {} | |
for p in preds: | |
outputs[p["label"]] = p["score"] | |
return outputs | |
gradio_app = gr.Interface( | |
fn=classify_audio, inputs=gr.Audio(type="filepath"), outputs=gr.Label() | |
) | |
if __name__ == "__main__": | |
gradio_app.launch(debug=True) |