File size: 605 Bytes
a9b52e3 074b1c4 50c57e9 a9b52e3 50c57e9 074b1c4 a9b52e3 074b1c4 a9b52e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import gradio as gr
from pyannote_viewer import PyannoteViewer
from pyannote.audio import Pipeline
import os
def apply_pipeline(audio: str) -> tuple:
pipeline = Pipeline.from_pretrained(
"pyannote/speech-separation-ami-1.0", use_auth_token=os.environ["HF_TOKEN"]
)
return pipeline(audio)
with gr.Blocks() as demo:
audio = gr.Audio(type="filepath")
btn = gr.Button("Apply separation pipeline")
pyannote_viewer = PyannoteViewer(interactive=False)
btn.click(fn=apply_pipeline, inputs=[audio], outputs=[pyannote_viewer])
if __name__ == "__main__":
demo.launch()
|