Spaces:
Runtime error
Runtime error
Commit
·
bf72a19
1
Parent(s):
b4fc8ba
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import gradio as gr
|
|
| 5 |
import pytube as pt
|
| 6 |
from speechbox import ASRDiarizationPipeline
|
| 7 |
|
| 8 |
-
MODEL_NAME = "openai/whisper-
|
| 9 |
|
| 10 |
device = 0 if torch.cuda.is_available() else "cpu"
|
| 11 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
@@ -69,15 +69,15 @@ mf_transcribe = gr.Interface(
|
|
| 69 |
title="Whisper Speaker Diarization: Transcribe Audio",
|
| 70 |
description=(
|
| 71 |
"Transcribe audio files with speaker diarization using [🤗 Speechbox](https://github.com/huggingface/speechbox/). "
|
| 72 |
-
"Demo uses the pre-trained checkpoint [Whisper
|
| 73 |
"transcriptions and [pyannote.audio](https://huggingface.co/pyannote/speaker-diarization) to label the speakers."
|
| 74 |
"\n\n"
|
| 75 |
"Check out the repo here: https://github.com/huggingface/speechbox/"
|
| 76 |
),
|
| 77 |
-
examples=[
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
],
|
| 81 |
allow_flagging="never",
|
| 82 |
)
|
| 83 |
|
|
|
|
| 5 |
import pytube as pt
|
| 6 |
from speechbox import ASRDiarizationPipeline
|
| 7 |
|
| 8 |
+
MODEL_NAME = "openai/whisper-small"
|
| 9 |
|
| 10 |
device = 0 if torch.cuda.is_available() else "cpu"
|
| 11 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
|
|
| 69 |
title="Whisper Speaker Diarization: Transcribe Audio",
|
| 70 |
description=(
|
| 71 |
"Transcribe audio files with speaker diarization using [🤗 Speechbox](https://github.com/huggingface/speechbox/). "
|
| 72 |
+
"Demo uses the pre-trained checkpoint [Whisper Small](https://huggingface.co/openai/whisper-small) for the ASR "
|
| 73 |
"transcriptions and [pyannote.audio](https://huggingface.co/pyannote/speaker-diarization) to label the speakers."
|
| 74 |
"\n\n"
|
| 75 |
"Check out the repo here: https://github.com/huggingface/speechbox/"
|
| 76 |
),
|
| 77 |
+
#examples=[
|
| 78 |
+
# ["./processed.wav", True],
|
| 79 |
+
# ["./processed.wav", False],
|
| 80 |
+
#],
|
| 81 |
allow_flagging="never",
|
| 82 |
)
|
| 83 |
|