tykiww commited on
Commit
e29651b
1 Parent(s): 18a0c49

Update services/diarization.py

Browse files
Files changed (1) hide show
  1. 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"""