diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py index 2c544fc..5da7048 100644 --- a/faster_whisper/transcribe.py +++ b/faster_whisper/transcribe.py @@ -36,12 +36,6 @@ class Segment(NamedTuple): no_speech_prob: float -class AudioInfo(NamedTuple): - language: str - language_probability: float - duration: float - - class TranscriptionOptions(NamedTuple): beam_size: int best_of: int @@ -63,6 +57,13 @@ class TranscriptionOptions(NamedTuple): append_punctuations: str +class AudioInfo(NamedTuple): + language: str + language_probability: float + duration: float + transcription_options: TranscriptionOptions + + class WhisperModel: def __init__( self, @@ -321,6 +322,7 @@ class WhisperModel: language=language, language_probability=language_probability, duration=duration, + transcription_options=options, ) return segments, audio_info