213d55dbdaa82a8d5edeede2a821fe2163881eb7,transcribe.py,,,#,41

Before Change

if __name__ == "__main__":
    arg_parser = argparse.ArgumentParser(description="DeepSpeech transcription")
    arg_parser = add_inference_args(arg_parser)
    arg_parser.add_argument("--audio-path",
                            default="audio.wav",
                            help="Audio file to predict on")
    arg_parser.add_argument("--offsets",
                            dest="offsets",
                            action="store_true",
                            help="Returns time offset information")
    arg_parser = add_decoder_args(arg_parser)
    args = arg_parser.parse_args()
    device = torch.device("cuda" if args.cuda else "cpu")
    model = load_model(device, args.model_path, args.half)

    decoder = load_decoder(decoder_type=args.decoder,
                           labels=model.labels,
                           lm_path=args.lm_path,
                           alpha=args.alpha,
                           beta=args.beta,
                           cutoff_top_n=args.cutoff_top_n,
                           cutoff_prob=args.cutoff_prob,
                           beam_width=args.beam_width,
                           lm_workers=args.lm_workers)

    spect_parser = SpectrogramParser(audio_conf=model.audio_conf,
                                     normalize=True)

    decoded_output, decoded_offsets = transcribe(audio_path=args.audio_path,

After Change

    transcribe(cfg=cfg)


if __name__ == "__main__":
    hydra_main()

Instances