Spaces:

tykiww
/

diarize_and_transcribe

Sleeping

tykiww commited on Aug 22, 2024

Commit

e29651b

verified ·

1 Parent(s): 18a0c49

Update services/diarization.py

Files changed (1) hide show

services/diarization.py CHANGED Viewed

@@ -17,6 +17,9 @@ class Diarizer:
             use_auth_token=os.environ["HUGGINGFACE_TOKEN"]
         )
         return pipeline
     def add_device(self, pipeline):
         """Offloaded to allow for best timing when working with GPUs"""

             use_auth_token=os.environ["HUGGINGFACE_TOKEN"]
         )
         return pipeline
+    def get_pipeline(self):
+        return self.pipeline
     def add_device(self, pipeline):
         """Offloaded to allow for best timing when working with GPUs"""