Skip to content

Commit 3fab826

Browse files
authored
Update whisper unit test (#258)
1 parent 9c449c1 commit 3fab826

File tree

1 file changed

+34
-30
lines changed

1 file changed

+34
-30
lines changed

tests/pipelines.test.js

Lines changed: 34 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -733,7 +733,7 @@ describe('Pipelines', () => {
733733
'openai/whisper-tiny.en', // English-only
734734
'openai/whisper-small', // Multilingual
735735
['openai/whisper-tiny.en', 'output_attentions'], // English-only + `output_attentions`
736-
['openai/whisper-base', 'output_attentions'], // Multilingual + `output_attentions`
736+
['openai/whisper-small', 'output_attentions'], // Multilingual + `output_attentions`
737737

738738
// wav2vec2
739739
'jonatasgrosman/wav2vec2-large-xlsr-53-english',
@@ -846,36 +846,40 @@ describe('Pipelines', () => {
846846
{ // Transcribe Japanese w/ word-level timestamps.
847847
let output = await transcriber(audioData, { return_timestamps: 'word', language: 'japanese', task: 'transcribe' });
848848
const target = {
849-
"text": "森長の美味しい牛乳は濃い青い牛乳ビーンを足らった階のパック牛乳である",
849+
"text": "モリナガの美味しい牛乳は濃い青色に牛乳瓶を払ったゼザインのパック牛乳である。",
850850
"chunks": [
851-
{ "text": "森", "timestamp": [0.14, 0.64] },
852-
{ "text": "長", "timestamp": [0.64, 0.82] },
853-
{ "text": "の", "timestamp": [0.82, 1.04] },
854-
{ "text": "美味", "timestamp": [1.04, 1.2] },
855-
{ "text": "しい", "timestamp": [1.2, 1.5] },
856-
{ "text": "牛", "timestamp": [1.5, 1.68] },
857-
{ "text": "乳", "timestamp": [1.68, 1.92] },
858-
{ "text": "は", "timestamp": [1.92, 2.14] },
859-
{ "text": "濃", "timestamp": [2.14, 2.32] },
860-
{ "text": "い", "timestamp": [2.32, 2.44] },
861-
{ "text": "青", "timestamp": [2.44, 2.66] },
862-
{ "text": "い", "timestamp": [2.66, 2.76] },
863-
{ "text": "牛", "timestamp": [2.76, 3.06] },
864-
{ "text": "乳", "timestamp": [3.06, 3.36] },
865-
{ "text": "ビ", "timestamp": [3.36, 3.58] },
866-
{ "text": "ーン", "timestamp": [3.58, 3.66] },
867-
{ "text": "を", "timestamp": [3.66, 3.82] },
868-
{ "text": "足", "timestamp": [3.82, 4] },
869-
{ "text": "ら", "timestamp": [4, 4.12] },
870-
{ "text": "った", "timestamp": [4.12, 4.3] },
871-
{ "text": "階", "timestamp": [4.3, 4.56] },
872-
{ "text": "の", "timestamp": [4.56, 4.92] },
873-
{ "text": "パ", "timestamp": [4.92, 5.1] },
874-
{ "text": "ック", "timestamp": [5.1, 5.2] },
875-
{ "text": "牛", "timestamp": [5.2, 5.44] },
876-
{ "text": "乳", "timestamp": [5.44, 5.64] },
877-
{ "text": "で", "timestamp": [5.64, 5.84] },
878-
{ "text": "ある", "timestamp": [5.84, 6.06] }
851+
{ "text": "モ", "timestamp": [0, 0.56] },
852+
{ "text": "リ", "timestamp": [0.56, 0.64] },
853+
{ "text": "ナ", "timestamp": [0.64, 0.8] },
854+
{ "text": "ガ", "timestamp": [0.8, 0.88] },
855+
{ "text": "の", "timestamp": [0.88, 1.04] },
856+
{ "text": "美味", "timestamp": [1.04, 1.22] },
857+
{ "text": "しい", "timestamp": [1.22, 1.46] },
858+
{ "text": "牛", "timestamp": [1.46, 1.76] },
859+
{ "text": "乳", "timestamp": [1.76, 1.94] },
860+
{ "text": "は", "timestamp": [1.94, 2.14] },
861+
{ "text": "濃", "timestamp": [2.14, 2.34] },
862+
{ "text": "い", "timestamp": [2.34, 2.48] },
863+
{ "text": "青", "timestamp": [2.48, 2.62] },
864+
{ "text": "色", "timestamp": [2.62, 2.84] },
865+
{ "text": "に", "timestamp": [2.84, 3] },
866+
{ "text": "牛", "timestamp": [3, 3.22] },
867+
{ "text": "乳", "timestamp": [3.22, 3.42] },
868+
{ "text": "瓶", "timestamp": [3.42, 3.58] },
869+
{ "text": "を", "timestamp": [3.58, 3.82] },
870+
{ "text": "払", "timestamp": [3.82, 4] },
871+
{ "text": "った", "timestamp": [4, 4.32] },
872+
{ "text": "ゼ", "timestamp": [4.32, 4.56] },
873+
{ "text": "ザ", "timestamp": [4.56, 4.6] },
874+
{ "text": "イ", "timestamp": [4.6, 4.74] },
875+
{ "text": "ン", "timestamp": [4.74, 4.8] },
876+
{ "text": "の", "timestamp": [4.8, 4.94] },
877+
{ "text": "パ", "timestamp": [4.94, 5.12] },
878+
{ "text": "ック", "timestamp": [5.12, 5.26] },
879+
{ "text": "牛", "timestamp": [5.26, 5.52] },
880+
{ "text": "乳", "timestamp": [5.52, 5.72] },
881+
{ "text": "で", "timestamp": [5.72, 5.86] },
882+
{ "text": "ある。", "timestamp": [5.86, 6.62] }
879883
]
880884
}
881885

0 commit comments

Comments
 (0)