Skip to content

Commit 418d519

Browse files
committed
Add Whisper::Context#full_get_segment
1 parent 324b8d4 commit 418d519

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

bindings/ruby/ext/ruby_whisper_context.c

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -521,6 +521,12 @@ ruby_whisper_full_get_segment_no_speech_prob(VALUE self, VALUE i_segment)
521521

522522
// High level API
523523

524+
static VALUE
525+
ruby_whisper_full_get_segment(VALUE self, VALUE i_segment)
526+
{
527+
return rb_whisper_segment_initialize(self, NUM2INT(i_segment));
528+
}
529+
524530
/*
525531
* Yields each Whisper::Segment:
526532
*
@@ -600,6 +606,7 @@ init_ruby_whisper_context(VALUE *mWhisper)
600606
rb_define_method(cContext, "full_parallel", ruby_whisper_full_parallel, -1);
601607

602608
// High leve
609+
rb_define_method(cContext, "full_get_segment", ruby_whisper_full_get_segment, 1);
603610
rb_define_method(cContext, "each_segment", ruby_whisper_each_segment, 0);
604611

605612
rb_define_method(cContext, "model", ruby_whisper_get_model, 0);

0 commit comments

Comments
 (0)