Merge pull request #183 from jhj0517/feature/integrate-whisperx

Fix strip bug
jhj0517 · Jun 27, 2024 · fbdb0fb · fbdb0fb
2 parents 8449dc3 + be3301d
commit fbdb0fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/diarize/diarizer.py b/modules/diarize/diarizer.py
@@ -7,6 +7,7 @@
 from modules.diarize.diarize_pipeline import DiarizationPipeline, assign_word_speakers
 from modules.diarize.audio_loader import load_audio
 
+
 class Diarizer:
     def __init__(self,
                  model_dir: str = os.path.join("models", "Diarization")
@@ -67,7 +68,7 @@ def run(self,
             speaker = "None"
             if "speaker" in segment:
                 speaker = segment["speaker"]
-            segment["text"] = speaker + "|" + segment["text"][1:]
+            segment["text"] = speaker + "|" + segment["text"].strip()
 
         elapsed_time = time.time() - start_time
         return diarized_result["segments"], elapsed_time