Spaces:
Runtime error
Runtime error
Commit
·
95736ea
1
Parent(s):
c60d983
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-small"
|
| 9 |
|
| 10 |
device = 0 if torch.cuda.is_available() else "cpu"
|
| 11 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
@@ -67,7 +67,7 @@ mf_transcribe = gr.Interface(
|
|
| 67 |
title="Whisper Speaker Diarization: Transcribe Audio",
|
| 68 |
description=(
|
| 69 |
"Transcribe audio files with speaker diarization using [🤗 Speechbox](https://github.com/huggingface/speechbox/). "
|
| 70 |
-
"Demo uses the pre-trained checkpoint [Whisper Small](https://huggingface.co/openai/whisper-small) for the ASR "
|
| 71 |
"transcriptions and [pyannote.audio](https://huggingface.co/pyannote/speaker-diarization) to label the speakers."
|
| 72 |
"\n\n"
|
| 73 |
"Check out the repo here: https://github.com/huggingface/speechbox/"
|
|
|
|
| 5 |
import pytube as pt
|
| 6 |
from speechbox import ASRDiarizationPipeline
|
| 7 |
|
| 8 |
+
MODEL_NAME = "openai/whisper-small.en"
|
| 9 |
|
| 10 |
device = 0 if torch.cuda.is_available() else "cpu"
|
| 11 |
HF_TOKEN = os.environ.get("HF_TOKEN")
|
|
|
|
| 67 |
title="Whisper Speaker Diarization: Transcribe Audio",
|
| 68 |
description=(
|
| 69 |
"Transcribe audio files with speaker diarization using [🤗 Speechbox](https://github.com/huggingface/speechbox/). "
|
| 70 |
+
"Demo uses the pre-trained checkpoint [Whisper Small.en](https://huggingface.co/openai/whisper-small.en) for the ASR "
|
| 71 |
"transcriptions and [pyannote.audio](https://huggingface.co/pyannote/speaker-diarization) to label the speakers."
|
| 72 |
"\n\n"
|
| 73 |
"Check out the repo here: https://github.com/huggingface/speechbox/"
|