run.ipynb

{"cells": [{"cell_type": "markdown", "id": "302934307671667531413257853548643485645", "metadata": {}, "source": ["# Gradio Demo: dialogue_mock_diarization"]}, {"cell_type": "code", "execution_count": null, "id": "272996653310673477252411125948039410165", "metadata": {}, "outputs": [], "source": ["!pip install -q gradio "]}, {"cell_type": "code", "execution_count": null, "id": "288918539441861185822528903084949547379", "metadata": {}, "outputs": [], "source": ["import gradio as gr\n", "\n", "speakers = [\n", "    \"Speaker 1\",\n", "    \"Speaker 2\",\n", "]\n", "\n", "def format_speaker(speaker, text):\n", "    return f\"{speaker}: {text}\"\n", "\n", "def mock_diarization(audio):\n", "    return [\n", "        {\n", "            \"speaker\": \"Speaker 1\",\n", "            \"text\": \"Hello, how are you?\",\n", "        },\n", "        {\n", "            \"speaker\": \"Speaker 2\",\n", "            \"text\": \"I'm fine, thank you!\",\n", "        },\n", "        {\n", "            \"speaker\": \"Speaker 1\",\n", "            \"text\": \"What's your name?\",\n", "        },\n", "        {\n", "            \"speaker\": \"Speaker 2\",\n", "            \"text\": \"My name is John Doe.\",\n", "        },\n", "        {\n", "            \"speaker\": \"Speaker 1\",\n", "            \"text\": \"Nice to meet you!\",\n", "        },\n", "        {\n", "            \"speaker\": \"Speaker 2\",\n", "            \"text\": \"Nice to meet you!\",\n", "        },\n", "    ]\n", "\n", "demo = gr.Interface(\n", "    fn=mock_diarization,\n", "    inputs=[gr.Audio(sources=[\"microphone\"])],\n", "    outputs=[gr.Dialogue(speakers=speakers, tags=None, formatter=format_speaker)],\n", "    title=\"Mock Speech Diarization\",\n", "    description=\"Mock speech diarization\",\n", "    api_name=\"predict\"\n", ")\n", "\n", "if __name__ == \"__main__\":\n", "    demo.launch()\n"]}], "metadata": {}, "nbformat": 4, "nbformat_minor": 5}