Commit Graph
44 Commits
Author SHA1 Message Date
Guillaume KleinandGitHub e94711bb5c Add property WhisperModel.supported_languages (#476)
* Expose function supported_languages

* Make it a method
2023-09-14 17:42:02 +02:00
Guillaume KleinandGitHub 0048844f54 Expose function available_models (#475)
* Expose function available_models

* Add test case
2023-09-14 17:17:01 +02:00
Guillaume KleinandGitHub 81086f6d33 Always run the encoder at the beginning of the loop (#468) 2023-09-12 14:44:37 +02:00
Guillaume KleinandGitHub f697945691 Update tokenizers requirement to include version 0.14 (#469) 2023-09-12 14:44:22 +02:00
Guillaume KleinandGitHub 727ab81f31 Improve error message for invalid task and language parameters (#466) 2023-09-12 10:02:23 +02:00
Guillaume KleinandGitHub 0285d46f6f Add more details about the requirements in the README (#463) 2023-09-08 14:35:17 +02:00
Guillaume KleinandGitHub 4a41746e55 Log a warning when the model is English-only but the language is set to something else (#454) 2023-09-04 11:55:40 +02:00
Guillaume KleinandGitHub 1e6eb967c9 Add "large" alias for "large-v2" model (#453) 2023-09-04 11:54:42 +02:00
Guillaume KleinandGitHub f0ff12965a Expose generation parameter no_repeat_ngram_size (#449) 2023-09-01 17:31:30 +02:00
Guillaume KleinandGitHub 5871858a5f Force the garbage collector to run after decoding the audio with PyAV (#448) 2023-09-01 15:25:13 +02:00
Guillaume KleinandGitHub 0f55c436fe Invalidate the cached encoder output when no_speech threshold is met (#376) 2023-07-24 10:57:15 +02:00
Guillaume KleinandGitHub 0e051a5b77 Prepend prefix tokens with the initial timestamp token (#358) 2023-07-18 15:22:39 +02:00
Guillaume KleinandGitHub c0d93d0829 Avoid computing higher temperatures on no_speech segments (#225)
Port commit https://github.com/openai/whisper/commit/e334ff141d5444fbf6904edaaf408e5b0b416fe8
2023-07-03 10:20:36 +02:00
Guillaume KleinandGitHub 19c294f978 Squash long words at window and sentence boundaries (#226)
Port commit https://github.com/openai/whisper/commit/255887f219e6b632bc1a6aac1caf28eecfca1bac
2023-07-03 10:20:20 +02:00
Guillaume KleinandGitHub efc4f61d85 Do not specify the vocabulary file extension in the download pattern (#311) 2023-06-20 10:53:11 +02:00
Guillaume KleinandGitHub a150adcc19 Enable onnxruntime dependency for Python 3.11 (#260) 2023-05-24 16:07:54 +02:00
Guillaume KleinandGitHub cf7c021573 Export __version__ at the module level (#258) 2023-05-24 15:50:37 +02:00
Guillaume KleinandGitHub 4db549b800 Make get_speech_timestamps backward compatible with the previous usage (#259) 2023-05-24 15:49:36 +02:00
Guillaume KleinandGitHub c99feb22dc Include requirements files in sdist (#240) 2023-05-24 12:55:15 +02:00
Guillaume KleinandGitHub 723cb97483 Fix occasional IndexError on empty segments (#227) 2023-05-24 12:55:04 +02:00
Guillaume KleinandGitHub 6a2da9a95c Also catch client-side network exceptions when synchronizing models (#228) 2023-05-11 15:07:15 +02:00
Guillaume KleinandGitHub 6a1d331d66 Add CONTRIBUTING.md (#229) 2023-05-11 15:06:46 +02:00
Guillaume KleinandGitHub 8340e04dc6 Assign words to the speech chunk with the greatest coverage (#180) 2023-04-25 15:54:31 +02:00
Guillaume KleinandGitHub 8cf5d5a4b3 Increase the default value of speech_pad_ms to 400 ms (#179) 2023-04-25 15:54:22 +02:00
Guillaume KleinandGitHub e06511f96b Rename AudioInfo to TranscriptionInfo (#174) 2023-04-24 16:29:17 +02:00
Guillaume KleinandGitHub 3adcc12d0f Clarify that the returned segments value is a generator (#144)
* Clarify that the returned segments value is a generator

* Update README.md
2023-04-13 09:50:53 +02:00
Guillaume KleinandGitHub e9a082dcf2 Keep segment timestamps aligned with words timestamps after VAD (#119) 2023-04-06 11:54:40 +02:00
Guillaume KleinandGitHub 051b3350e5 Add some info and debug logs (#113) 2023-04-05 16:57:59 +02:00
Guillaume KleinandGitHub a5d03e55fa Prevent out of range error in method split_tokens_on_unicode (#111) 2023-04-04 10:51:14 +02:00
Guillaume KleinandGitHub 2f266eb844 Fix VAD index error when a predicted timestamps is too large (#107) 2023-04-03 19:34:54 +02:00
Guillaume KleinandGitHub 19698c95f8 Support VAD filter (#95)
* Support VAD filter

* Generalize function collect_samples

* Define AudioSegment class

* Only pass prompt and prefix to the first chunk

* Add dict argument vad_parameters

* Fix isort format

* Rename method

* Update README

* Add shortcut when the chunk offset is 0

* Reword readme

* Fix end property

* Concatenate the speech chunks

* Cleanup diff

* Increase default speech pad

* Update README

* Increase default speech pad
2023-04-03 17:22:48 +02:00
Guillaume KleinandGitHub f20bb258de Support separating the left and right audio channels (#97) 2023-04-03 11:22:43 +02:00
Guillaume KleinandGitHub def70d8496 Update headings in the Usage section 2023-03-31 18:54:55 +02:00
Guillaume KleinandGitHub 8246479fda Ignore the invalid audio frames (#82) 2023-03-27 10:19:22 +02:00
Guillaume KleinandGitHub de7682a2f0 Automatically download converted models from the Hugging Face Hub (#70)
* Automatically download converted models from the Hugging Face Hub

* Remove unused import

* Remove non needed requirements in dev mode

* Remove extra index URL when pip install in CI

* Allow downloading to a specific directory

* Update docstring

* Add argument to disable the progess bars

* Fix typo in docstring
2023-03-24 10:55:55 +01:00
Guillaume KleinandGitHub 523ae2180f Run the encoder only once for each 30-second window (#73) 2023-03-24 10:53:49 +01:00
Guillaume KleinandGitHub 66efd02bd0 Run some automatic tests with GitHub Actions (#68) 2023-03-22 20:50:03 +01:00
Guillaume KleinandGitHub 8bd013ea99 Add word-level timestamps (#43)
* Add word-level timestamps

* Fix alignment between the segments and the lists of words

* Fix truncated words list when the replacement character is decoded

* Check for empty text_tokens

* Add usage example in the readme

* Update ctranslate2 to 3.9

* Skip empty segment

* Set typing for the new methods
2023-03-15 15:02:28 +01:00
Guillaume KleinandGitHub 873992623c Accept the audio waveform as an input to transcribe() (#21) 2023-02-28 19:01:31 +01:00
Guillaume KleinandGitHub 47a62ab975 Update README.md 2023-02-13 17:43:22 +01:00
Guillaume KleinandGitHub 269b3dfb10 Expose the device_index argument (#5) 2023-02-13 11:06:40 +01:00
Guillaume KleinandGitHub 7d1d0541c8 Add the initial_prompt parameter (#2)
* Add the initial_prompt parameter

* Add docstring
2023-02-12 11:42:21 +01:00
Guillaume KleinandGitHub 23d2d64259 Update transcribe.py 2023-02-11 11:47:07 +01:00
Guillaume KleinandGitHub c0ec7fe83b Update README.md 2023-02-11 11:46:09 +01:00