Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -483,7 +483,8 @@ class WhisperBase(ABC):
             output_path += '.txt'
 
         elif file_format == "csv":
-            content = get_csv(transcribed_segments)
+            params = WhisperParameters.as_value(*whisper_params)
+            content = get_csv(transcribed_segments,params.is_diarize)
             output_path += '.csv'
 
         write_file(content, output_path)