mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-06-02 16:05:35 +02:00
1f5fdbecb4
* Add VAD models * Extract function to normalize model path from ruby_whisper_initialize() * Define ruby_whisper_vad_params struct * Add VAD-related features to Whisper::Params * Add tests for VAD-related features * Define Whisper::VADParams * Add Whisper::VAD::Params attributes * Add test suite for VAD::Params * Make older test to follow namespace change * Add test for transcription with VAD * Add assertion for test_vad_params * Add signatures for VAD-related methods * Define VAD::Params#== * Add test for VAD::Params#== * Fix Params#vad_params * Add test for Params#vad_params * Fix signature of Params#vad_params * Use macro to define VAD::Params params * Define VAD::Params#initialize * Add tests for VAD::Params#initialize * Add signature for VAD::Params.new * Add documentation on VAD in README * Wrap register_callbask in prepare_transcription for clear meanings * Set whisper_params.vad_params just before transcription * Don't touch NULL * Define ruby_whisper_params_type * Use TypedData_XXX for ruby_whisper_params instead of Data_XXX * Remove unused functions * Define rb_whisper_model_data_type * Use TypedData_XXX for ruby_whisper_model instead of Data_XXX * Define ruby_whisper_segment_type * Use TypedData_XXX for ruby_whisper_segment instead of Data_XXX * Define ruby_whisper_type * Use TypedData_XXX for ruby_whisper instead of Data_XXX * Qualify with const
20 lines
486 B
Ruby
20 lines
486 B
Ruby
require_relative "helper"
|
|
|
|
class TestVAD < TestBase
|
|
def setup
|
|
@whisper = Whisper::Context.new("base.en")
|
|
vad_params = Whisper::VAD::Params.new
|
|
@params = Whisper::Params.new(
|
|
vad: true,
|
|
vad_model_path: "silero-v5.1.2",
|
|
vad_params:
|
|
)
|
|
end
|
|
|
|
def test_transcribe
|
|
@whisper.transcribe(TestBase::AUDIO, @params) do |text|
|
|
assert_match(/ask not what your country can do for you[,.] ask what you can do for your country/i, text)
|
|
end
|
|
end
|
|
end
|