File size: 696 Bytes
3f44072
de18ebe
3f44072
de18ebe
0fe103c
4c8ad45
 
b66750f
3f44072
 
6c5a926
3f44072
 
b66750f
 
 
3f44072
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import gradio as gr
from pyannote.audio import Pipeline

def diarization(audio):
    pipeline = Pipeline.from_pretrained("pyannote/speaker-segmentation")
    #parameters = {"onset":0.7,"offset":0.3,"min_duration_on":0.0,"min_duration_off":0.0}
    #pipeline.instantiate(parameters)
    output = pipeline(audio)
    answer = ""
    for turn, _, speaker in output.itertracks(yield_label=True):
        answer += "{} said something starting from {:.2f} and ends on {:.2f}\n".format(speaker,turn.start,turn.end)
    return answer

app = gr.Interface(fn=diarization,
                inputs=gr.inputs.Audio(source="upload", type="filepath", label="audio"),
                outputs="text")
app.launch()