From 3ee50d43044ae1ffe919b94edc80c8a5cf3aa7ee Mon Sep 17 00:00:00 2001 From: BenAAndrew Date: Sat, 31 Jul 2021 14:13:11 +0100 Subject: [PATCH] Fix create dataset CLI --- dataset/create_dataset.py | 3 ++- dataset/transcribe.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/dataset/create_dataset.py b/dataset/create_dataset.py index e00d6ee..b14aa5c 100644 --- a/dataset/create_dataset.py +++ b/dataset/create_dataset.py @@ -8,6 +8,7 @@ sys.path.append(dirname(dirname(abspath(__file__)))) from dataset.audio_processing import convert_audio from dataset.clip_generator import clip_generator from dataset.analysis import save_dataset_info +from dataset.transcribe import create_transcription_model def create_dataset( @@ -82,4 +83,4 @@ if __name__ == "__main__": parser.add_argument("-i", "--info_path", help="Path to save info file", type=str, default="info.json") args = parser.parse_args() - create_dataset(**vars(args)) + create_dataset(**vars(args), transcription_model=create_transcription_model()) diff --git a/dataset/transcribe.py b/dataset/transcribe.py index aee982d..1e24836 100644 --- a/dataset/transcribe.py +++ b/dataset/transcribe.py @@ -118,5 +118,6 @@ if __name__ == "__main__": parser.add_argument("-i", "--input_path", help="Path to audio file", type=str, required=True) args = parser.parse_args() - text = transcribe(args.input_path) + model = create_transcription_model() + text = model.transcribe(args.input_path) print("Text: ", text)