Spaces:
Runtime error
Runtime error
sanchit-gandhi
commited on
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 |
|