Allow init OpenVINO encoder (#14)

Signed-off-by: Miguel Álvarez <miguelwork92@gmail.com>
GiviMAD · Nov 12, 2023 · 7f18ba6 · 7f18ba6
1 parent 86943e6
commit 7f18ba6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/pom.xml b/pom.xml
@@ -8,7 +8,7 @@
     <artifactId>whisper-jni</artifactId>
     <name>whisper-jni</name>
     <url>https://github.com/GiviMAD/whisper-jni</url>
-    <version>1.4.3-1</version>
+    <version>1.4.3-2</version>
     <description>A JNI wrapper for [whisper.cpp](https://github.com/ggerganov/whisper.cpp), allows to transcribe speech to text in Java</description>
 
     <licenses>

diff --git a/src/main/java/io/github/givimad/whisperjni/WhisperJNI.java b/src/main/java/io/github/givimad/whisperjni/WhisperJNI.java
@@ -5,7 +5,6 @@
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
-import java.util.function.Consumer;
 
 /**
  * The {@link WhisperJNI} class allows to use whisper.cpp thought the JNI.
@@ -122,6 +121,17 @@ public WhisperState initState(WhisperContext context) {
         return new WhisperState(this, initState(context.ref), context);
     }
 
+    /**
+     * Initializes OpenVino encoder.
+     *
+     * @param context a {@link WhisperContext} instance.
+     * @param device the device name.
+     */
+    public void initOpenVINO(WhisperContext context, String device) {
+        WhisperJNIPointer.assertAvailable(context);
+        initOpenVINOEncoder(context.ref, device);
+    }
+
     /**
      * Is multilingual.
      *