Spaces:
Running
Running
jhj0517
commited on
Commit
·
be3301d
1
Parent(s):
8449dc3
fix strip bug
Browse files
modules/diarize/diarizer.py
CHANGED
@@ -7,6 +7,7 @@ import logging
|
|
7 |
from modules.diarize.diarize_pipeline import DiarizationPipeline, assign_word_speakers
|
8 |
from modules.diarize.audio_loader import load_audio
|
9 |
|
|
|
10 |
class Diarizer:
|
11 |
def __init__(self,
|
12 |
model_dir: str = os.path.join("models", "Diarization")
|
@@ -67,7 +68,7 @@ class Diarizer:
|
|
67 |
speaker = "None"
|
68 |
if "speaker" in segment:
|
69 |
speaker = segment["speaker"]
|
70 |
-
segment["text"] = speaker + "|" + segment["text"]
|
71 |
|
72 |
elapsed_time = time.time() - start_time
|
73 |
return diarized_result["segments"], elapsed_time
|
|
|
7 |
from modules.diarize.diarize_pipeline import DiarizationPipeline, assign_word_speakers
|
8 |
from modules.diarize.audio_loader import load_audio
|
9 |
|
10 |
+
|
11 |
class Diarizer:
|
12 |
def __init__(self,
|
13 |
model_dir: str = os.path.join("models", "Diarization")
|
|
|
68 |
speaker = "None"
|
69 |
if "speaker" in segment:
|
70 |
speaker = segment["speaker"]
|
71 |
+
segment["text"] = speaker + "|" + segment["text"].strip()
|
72 |
|
73 |
elapsed_time = time.time() - start_time
|
74 |
return diarized_result["segments"], elapsed_time
|