游雁
2023-03-15 5b355e0f9340c3f645c417e4493fd08f3889c588
funasr/bin/modelscope_infer.py
@@ -17,7 +17,7 @@
                        help="model name in modelscope")
    parser.add_argument("--model_revision",
                        type=str,
                        default="v1.0.3",
                        default="v1.0.4",
                        help="model revision in modelscope")
    parser.add_argument("--local_model_path",
                        type=str,
@@ -82,6 +82,9 @@
            wav_id, wav_path = line.strip().split()
            logging.info("decoding, utt_id: ['{}']".format(wav_id))
            rec_result = inference_pipeline(audio_in=wav_path)
            text = rec_result["text"]
            if 'text' in rec_result:
                text = rec_result["text"]
            else:
                text = ''
            f_out.write(wav_id + " " + text + "\n")
            logging.info("best hypo: {} \n".format(text))