baguioni's picture
Update app.py
f150fa4
raw
history blame
1.19 kB
import gradio as gr
import ast
model = gr.Interface.load("huggingface/pyannote/voice-activity-detection")
def format_inference(output):
if output:
timestamps = []
for out in output:
timestamps.append(f"Start: {out['start']}s; Stop{out['stop']}s")
return "\n".join(timestamps)
else:
return "No voice activity detected."
def inference(audio_file):
output = model(audio_file)
output_list = ast.literal_eval(output)
return format_inference(output_list)
inputs = gr.inputs.Audio(label="Input Audio", type="filepath", source="microphone")
outputs = gr.outputs.Textbox(label="Voice timestamps", type="auto")
title = "Voice Activity Detection"
description = "Record or upload an audio file and detected voices will be timestamped."
examples = ['samples/talk.wav', 'samples/talk2.wav', 'samples/silence.wav', ]
article = "pyannote, https://github.com/pyannote/pyannote-audio"
gr.Interface(inference,
inputs,
outputs,
title=title,
description=description,
article=article,
examples=examples,
theme="grass").launch(debug=True)