whisper.cpp 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 561b7b824c1f631537d681fb1e5a1eafe20546a8468705ecd2ba3989e0062c69
-  data.tar.gz: e047a8ab099358095100a55cd30aca398bbe8b8c4c55cfb0713cdf33e87d2e1a
+  metadata.gz: 3d1b91cc9ac6e7682429f261d3e9b283c4c34ef246a2e88d07f1c294d145d1f4
+  data.tar.gz: be2a0ff1c4eda542685c209e6c64b7f3f3f902c4353534747922ea89c42ef24d
 SHA512:
-  metadata.gz: 80c60958d31df322ac570c97ca6d758278f8711d9bfff371be34e8159ffcadc2facf149aa4c7b55de49a2d2d8812b570d0ddd68e543172301f930fe494a4243b
-  data.tar.gz: 02fd686ff1857adf234c53fa0719035e797a82bed4b165e54004c0b7c903728eb7d8c44273b3f46e6ff21a32df44e29986092b72ccfd853324749e639fa303ed
+  metadata.gz: c992abd9e3e9b48d6b31cfa6489b5b09489781d251c2d3ca24d4932fca579b7d19b7483a1d72aefe77ac0655d5d5af19708bf6342cf6e5d4922d3663b1244acc
+  data.tar.gz: 8890fa1e7653f8690c3686a7751e597958c06adfaa3a0b3a3e74c947920f1da2014ada7eebe4fd78e85cc2a74fe235b244c43817937072ffe3b7c75792567ec1

data/lib/whisper/model.rb

@@ -8,6 +8,7 @@ module Whisper
     def initialize(model_path)
       @model_path = model_path
       @ctx = nil
+      init_whisper_context
     end
 
     def transcribe_from_file(audio_file_path, format: 'plaintext', **params)
@@ -17,8 +18,6 @@ module Whisper
     end
 
     def transcribe_from_audio_data(audio_data, format: 'plaintext', **params)
-      init_whisper_context(params)
-
       # Prepare full params
       full_params = default_full_params(params)
 
@@ -49,19 +48,13 @@ module Whisper
 
     private
 
-    def init_whisper_context(params)
+    def init_whisper_context params = {}
       return unless @ctx.nil?
 
       ctx_params = Whisper.whisper_context_default_params
 
-      # Set user-provided context params
-      user_ctx_params = params.fetch(:context_params, {})
-      user_ctx_params.each do |key, value|
-        if ctx_params.members.include?(field)
-          ctx_params[key] = value
-        else
-          warn "Unknown context_param field: #{field}"
-        end
+      params.select{ |k, _| ctx_params.members.include? k }.each do |key, value|
+        ctx_params[key] = value
       end
       ctx_params[:gpu_device] = ENV['WHISPER_GPU']&.to_i || 0
 
@@ -70,22 +63,18 @@ module Whisper
       raise 'Failed to initialize Whisper model' if @ctx.null?
     end
 
-    def default_full_params(params)
+    def default_full_params params = {}
       # Get default full params
       strategy = params.fetch(:sampling_strategy, Whisper::WHISPER_SAMPLING_GREEDY)
       full_params = Whisper.whisper_full_default_params(strategy)
 
-        # Set translate to false to prevent translation to English
+      # Set translate to false to prevent translation to English
       full_params[:translate] = false
+      full_params[:language]  = FFI::MemoryPointer.from_string 'auto'
 
       # Set user-provided full params
-      user_full_params = params.fetch(:full_params, {})
-      user_full_params.each do |key, value|
-        if full_params.members.include?(field)
-          full_params[key] = value
-        else
-          warn "Unknown full_param field: #{field}"
-        end
+      params.select{ |k, _| full_params.members.include? k }.each do |key, value|
+        full_params[key] = value
       end
 
       full_params

data/lib/whisper/version.rb

@@ -1,5 +1,5 @@
 module Whisper
 
-  VERSION = '0.3.1'
+  VERSION = '0.3.2'
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: whisper.cpp
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - Braulio Oliveira