mirror of
https://github.com/malarinv/jasper-asr.git
synced 2026-06-13 12:32:08 +00:00
refactored arg parsing to take server cli args
This commit is contained in:
22
jasper/transcribe.py
Normal file
22
jasper/transcribe.py
Normal file
@@ -0,0 +1,22 @@
|
||||
from pathlib import Path
|
||||
from .asr import JasperASR
|
||||
from .utils import arg_parser
|
||||
|
||||
|
||||
def main():
|
||||
parser = arg_parser('jasper_transcribe')
|
||||
parser.description = 'transcribe audio file to text'
|
||||
parser.add_argument(
|
||||
"audio_file",
|
||||
type=Path,
|
||||
help="audio file(16khz 1channel int16 wav) to transcribe",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--greedy", type=bool, default=False, help="enables greedy decoding"
|
||||
)
|
||||
args = parser.parse_args()
|
||||
args_dict = vars(args)
|
||||
audio_file = args_dict.pop("audio_file")
|
||||
greedy = args_dict.pop("greedy")
|
||||
jasper_asr = JasperASR(**args_dict)
|
||||
jasper_asr.transcribe_file(audio_file, greedy)
|
||||
Reference in New Issue
Block a user