Refactor whisper-processing.cpp to update duration_ms in run_whisper_…

…inference
locaal-ai · May 20, 2024 · d3764f5 · d3764f5
1 parent 69586b0
commit d3764f5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/whisper-utils/whisper-processing.cpp b/src/whisper-utils/whisper-processing.cpp
@@ -139,6 +139,8 @@ int run_whisper_inference(struct cleanstream_data *gf, const float *pcm32f_data,
 	// run the inference
 	int whisper_full_result = -1;
 	try {
+		gf->whisper_params.duration_ms =
+			(int)((float)pcm32f_size / WHISPER_SAMPLE_RATE * 1000.0f);
 		whisper_full_result = whisper_full(gf->whisper_context, gf->whisper_params,
 						   pcm32f_data, (int)pcm32f_size);
 	} catch (const std::exception &e) {