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/"
|