Update run.py

jan-grzybek-ampere · web-flow · commit aa8864946f27 · 2024-02-29T04:22:52.000+01:00
diff --git a/speech_recognition/whisper/run.py b/speech_recognition/whisper/run.py
@@ -23,7 +23,7 @@ def single_pass_pytorch(_runner, _librispeech):
         )
 
     def transcribe_wrapper(audio):
-        return transcribe(model, audio, verbose=None)
+        return transcribe(model, audio, no_speech_threshold=1.0, verbose=None)
 
     runner = PyTorchRunnerV2(transcribe_wrapper, throughput_only=True)
     librispeech = LibriSpeech()
@@ -52,7 +52,7 @@ def single_pass_pytorch(_runner, _librispeech):
         )
 
     def transcribe_wrapper(audio):
-        return transcribe(model, audio, verbose=None)
+        return transcribe(model, audio, no_speech_threshold=1.0, verbose=None)
 
     runner = PyTorchRunnerV2(transcribe_wrapper)
     librispeech = LibriSpeech()

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ def single_pass_pytorch(_runner, _librispeech):`
`23`	`23`	`)`
`24`	`24`
`25`	`25`	`def transcribe_wrapper(audio):`
`26`		`- return transcribe(model, audio, verbose=None)`
	`26`	`+ return transcribe(model, audio, no_speech_threshold=1.0, verbose=None)`
`27`	`27`
`28`	`28`	`runner = PyTorchRunnerV2(transcribe_wrapper, throughput_only=True)`
`29`	`29`	`librispeech = LibriSpeech()`
`@@ -52,7 +52,7 @@ def single_pass_pytorch(_runner, _librispeech):`
`52`	`52`	`)`
`53`	`53`
`54`	`54`	`def transcribe_wrapper(audio):`
`55`		`- return transcribe(model, audio, verbose=None)`
	`55`	`+ return transcribe(model, audio, no_speech_threshold=1.0, verbose=None)`
`56`	`56`
`57`	`57`	`runner = PyTorchRunnerV2(transcribe_wrapper)`
`58`	`58`	`librispeech = LibriSpeech()`