Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4a5aaa8
1
Parent(s):
474e9f2
send pipeline to GPU
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import os
|
|
| 7 |
def apply_pipeline(audio):
|
| 8 |
"""Apply specified pipeline on the indicated audio file"""
|
| 9 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1", use_auth_token=os.environ["HF_TOKEN"])
|
|
|
|
| 10 |
annotations = pipeline(audio)
|
| 11 |
|
| 12 |
return ((audio, annotations), (audio, annotations))
|
|
|
|
| 7 |
def apply_pipeline(audio):
|
| 8 |
"""Apply specified pipeline on the indicated audio file"""
|
| 9 |
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1", use_auth_token=os.environ["HF_TOKEN"])
|
| 10 |
+
pipeline.to("cuda")
|
| 11 |
annotations = pipeline(audio)
|
| 12 |
|
| 13 |
return ((audio, annotations), (audio, annotations))
|