'#1909: Use a smaller buffer to avoid Vosk slow down.

sepinf-inc · Sep 30, 2023 · df087de · df087de
1 parent d71c6c8
commit df087de
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/iped-engine/src/main/java/iped/engine/task/transcript/VoskTranscriptTask.java b/iped-engine/src/main/java/iped/engine/task/transcript/VoskTranscriptTask.java
@@ -97,7 +97,8 @@ protected TextAndScore transcribeAudio(File tmpFile) throws Exception {
             int words = 0;
 
             int nbytes;
-            byte[] buf = new byte[1 << 20];
+            // Buffer must be small (see #1909)
+            byte[] buf = new byte[(int) Math.min(tmpFile.length(), 1 << 16)];
             while ((nbytes = ais.read(buf)) >= 0) {
                 if (recognizer.acceptWaveForm(buf, nbytes)) {
                     TextScoreWords result = decodeFromJson(recognizer.getResult());