Spaces:
Runtime error
Runtime error
Commit ·
7104546
1
Parent(s): f1bf51a
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,8 +21,8 @@ from transformers.pipelines.audio_utils import ffmpeg_read
|
|
| 21 |
HF_TOKEN = "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"
|
| 22 |
# set up the diarization pipeline
|
| 23 |
diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.0", use_auth_token=HF_TOKEN)
|
| 24 |
-
|
| 25 |
-
|
| 26 |
|
| 27 |
def transcribe(audio_path, task="transcribe", group_by_speaker=True, progress=gr.Progress()):
|
| 28 |
# # run Whisper JAX asynchronously using Gradio client (endpoint)
|
|
|
|
| 21 |
HF_TOKEN = "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"
|
| 22 |
# set up the diarization pipeline
|
| 23 |
diarization_pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.0", use_auth_token=HF_TOKEN)
|
| 24 |
+
if torch.cuda.is_available():
|
| 25 |
+
diarization_pipeline.to(torch.device("cuda"))
|
| 26 |
|
| 27 |
def transcribe(audio_path, task="transcribe", group_by_speaker=True, progress=gr.Progress()):
|
| 28 |
# # run Whisper JAX asynchronously using Gradio client (endpoint)
|