Spaces:
Sleeping
Sleeping
Update services/diarization.py
Browse files- services/diarization.py +3 -0
services/diarization.py
CHANGED
@@ -17,6 +17,9 @@ class Diarizer:
|
|
17 |
use_auth_token=os.environ["HUGGINGFACE_TOKEN"]
|
18 |
)
|
19 |
return pipeline
|
|
|
|
|
|
|
20 |
|
21 |
def add_device(self, pipeline):
|
22 |
"""Offloaded to allow for best timing when working with GPUs"""
|
|
|
17 |
use_auth_token=os.environ["HUGGINGFACE_TOKEN"]
|
18 |
)
|
19 |
return pipeline
|
20 |
+
|
21 |
+
def get_pipeline(self):
|
22 |
+
return self.pipeline
|
23 |
|
24 |
def add_device(self, pipeline):
|
25 |
"""Offloaded to allow for best timing when working with GPUs"""
|