From 12c1ac8b57a2b09e9340a49670c22fc89b26ea04 Mon Sep 17 00:00:00 2001 From: Raivis Dejus Date: Sat, 17 Aug 2024 18:48:06 +0300 Subject: [PATCH] Fix for extra spaces in whisper.cpp transcripts (#890) --- buzz/transcriber/file_transcriber.py | 3 +++ tests/transcriber/whisper_cpp_file_transcriber_test.py | 6 +++--- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/buzz/transcriber/file_transcriber.py b/buzz/transcriber/file_transcriber.py index ef0f4ec29..792922257 100644 --- a/buzz/transcriber/file_transcriber.py +++ b/buzz/transcriber/file_transcriber.py @@ -62,6 +62,9 @@ def run(self): self.error.emit(str(exc)) return + for segment in segments: + segment.text = segment.text.strip() + self.completed.emit(segments) for ( diff --git a/tests/transcriber/whisper_cpp_file_transcriber_test.py b/tests/transcriber/whisper_cpp_file_transcriber_test.py index bd9a993eb..9683ba0f5 100644 --- a/tests/transcriber/whisper_cpp_file_transcriber_test.py +++ b/tests/transcriber/whisper_cpp_file_transcriber_test.py @@ -25,7 +25,7 @@ class TestWhisperCppFileTranscriber: False, [Segment(0, 6560, "Bienvenue dans Passe-Relle. Un podcast pensé pour")], ), - (True, [Segment(30, 740, "Bienvenue"), Segment(740, 1070, " dans")]), + (True, [Segment(30, 740, "Bienvenue"), Segment(740, 1070, "dans")]), ], ) def test_transcribe( @@ -81,9 +81,9 @@ def test_transcribe( [ ( False, - [Segment(0, 7000, " Mani uzstrauts, laikabstākļi, tapēc uz jūru, es diezvajī braukša.")], + [Segment(0, 7000, "Mani uzstrauts, laikabstākļi, tapēc uz jūru, es diezvajī braukša.")], ), - (True, [Segment(380, 500, " Mani"), Segment(500, 1880, " uzstrauts,"), Segment(1880, 3920, " laikabstākļi")]), + (True, [Segment(380, 500, "Mani"), Segment(500, 1880, "uzstrauts,"), Segment(1880, 3920, "laikabstākļi")]), ], ) # Problematic part is in "laikabstākļi" where "ļ" gets returned from whisper.cpp in two segments