LLM/WhisperLiveKit
1
0
Fork 0
mirror of https://github.com/QuentinFuxa/WhisperLiveKit.git synced 2026-03-07 22:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

refacto 0

Browse Source
This commit is contained in:
Quentin Fuxa
2025-03-18 18:31:23 +01:00
parent 4f87ac3ea4
commit dc02bcdbdd
7 changed files with 576 additions and 408 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
diarization/diarization_online.py
View File

@@ -5,7 +5,7 @@ import numpy as np
import logging
from diart import SpeakerDiarization
from diart import SpeakerDiarization, SpeakerDiarizationConfig
from diart.inference import StreamingInference
from diart.sources import AudioSource
from timed_objects import SpeakerSegment
@@ -103,8 +103,8 @@ class WebSocketAudioSource(AudioSource):
class DiartDiarization:
def __init__(self, sample_rate: int, use_microphone: bool = False):
self.pipeline = SpeakerDiarization()
def __init__(self, sample_rate: int, config : SpeakerDiarizationConfig = None, use_microphone: bool = False):
self.pipeline = SpeakerDiarization(config=config)
self.observer = DiarizationObserver()
if use_microphone: