chore(transcription): use short video for local model test

larriereguichet · Jun 12, 2024 · 256cfc6 · 256cfc6
1 parent 501217b
commit 256cfc6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/tests/src/transcription/whisper/transcriber/openai-transcriber.spec.ts b/packages/tests/src/transcription/whisper/transcriber/openai-transcriber.spec.ts
@@ -78,9 +78,9 @@ describe('Open AI Whisper transcriber', function () {
   })
 
   it('May transcribe a media file using a local PyTorch model', async function () {
-    this.timeout(11 * 1000 * 60)
+    this.timeout(2 * 1000 * 60)
     await transcriber.transcribe({
-      mediaFilePath: frVideoPath,
+      mediaFilePath: shortVideoPath,
       model: await TranscriptionModel.fromPath(join(modelsDirectory, 'tiny.pt')),
       language: 'en'
     })

diff --git a/packages/tests/src/transcription/whisper/transcriber/timestamped-transcriber.spec.ts b/packages/tests/src/transcription/whisper/transcriber/timestamped-transcriber.spec.ts
@@ -73,7 +73,7 @@ describe('Linto timestamped Whisper transcriber', function () {
   it('May transcribe a media file using a local PyTorch model file', async function () {
     this.timeout(2 * 1000 * 60)
     await transcriber.transcribe({
-      mediaFilePath: frVideoPath,
+      mediaFilePath: shortVideoPath,
       model: await TranscriptionModel.fromPath(join(modelsDirectory, 'tiny.pt')),
       language: 'en'
     })