Fix

src/whisper_ctranslate2/diarization.py

@@ -7,7 +7,6 @@
 from faster_whisper.audio import decode_audio
 
 
-
 class Diarization:
     def __init__(
         self,
@@ -22,9 +21,7 @@ def __init__(
             model_name, use_auth_token=use_auth_token
         ).to(device)
 
-    def __call__(
-        self, audio: str, min_speakers=None, max_speakers=None
-    ):
+    def __call__(self, audio: str, min_speakers=None, max_speakers=None):
         audio = decode_audio(audio)
         audio_data = {
             "waveform": torch.from_numpy(audio[None, :]),

src/whisper_ctranslate2/whisper_ctranslate2.py

@@ -560,7 +560,7 @@ def main():
         print(f"Time used for inference: {datetime.datetime.now() - start_time}")
 
         start_time = datetime.datetime.now()
-        
+
         if len(hf_token) > 0:
             diarize_segments = diarize_model(
                 audio_path, min_speakers=None, max_speakers=None

src/whisper_ctranslate2/writers.py

@@ -68,7 +68,6 @@ def iterate_subtitles():
             subtitle: list[dict] = []
             last = result["segments"][0]["words"][0]["start"]
             for segment in result["segments"]:
-
                 if "speaker" in segment:
                     speaker = f"[{segment['speaker']}]: "
                 else:
@@ -106,13 +105,11 @@ def iterate_subtitles():
             if len(subtitle) > 0:
                 yield subtitle
 
-
-#        if "speaker" in segment:
-#            segment_text = f"[{segment['speaker']}]: " + segment["text"].strip().replace("-->", "->")
-#        else:
-#            segment_text = segment["text"].strip().replace("-->", "->")
+        #        if "speaker" in segment:
+        #            segment_text = f"[{segment['speaker']}]: " + segment["text"].strip().replace("-->", "->")
+        #        else:
+        #            segment_text = segment["text"].strip().replace("-->", "->")
 
-
         if (
             len(result["segments"]) > 0
             and "words" in result["segments"][0]