'#1823: convert UI language to whisper supported language format

sepinf-inc · Apr 13, 2024 · 71e125a · 71e125a
1 parent 147cdf1
commit 71e125a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/iped-engine/src/main/java/iped/engine/task/transcript/WhisperTranscriptTask.java b/iped-engine/src/main/java/iped/engine/task/transcript/WhisperTranscriptTask.java
@@ -40,7 +40,12 @@ protected Server startServer0(int device) throws IOException {
         int cpus = getNumProcessors();
         int threads = Runtime.getRuntime().availableProcessors() / cpus;
 
-        pb.command(python, script, model, Integer.toString(device), Integer.toString(threads), transcriptConfig.getLanguages().get(0));
+        String lang = transcriptConfig.getLanguages().get(0);
+        if (lang.contains("-")) {
+            lang = lang.substring(0, lang.indexOf("-"));
+        }
+
+        pb.command(python, script, model, Integer.toString(device), Integer.toString(threads), lang);
 
         Process process = pb.start();