jhj0517 commited on
Commit
be3301d
·
1 Parent(s): 8449dc3

fix strip bug

Browse files
Files changed (1) hide show
  1. modules/diarize/diarizer.py +2 -1
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"][1:]
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