mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-04 14:44:18 +00:00
Update test_onnx.py
This commit is contained in:
parent
fefae1f8df
commit
64903abf77
@ -28,10 +28,11 @@ Use the onnx model to generate a wav:
|
||||
|
||||
import argparse
|
||||
import logging
|
||||
from pathlib import Path
|
||||
|
||||
import onnxruntime as ort
|
||||
import torch
|
||||
import torchaudio
|
||||
|
||||
from tokenizer import Tokenizer
|
||||
|
||||
|
||||
@ -47,6 +48,12 @@ def get_parser():
|
||||
help="Path to the onnx model.",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--speakers",
|
||||
type=Path,
|
||||
default=Path("data/speakers.txt"),
|
||||
help="Path to speakers.txt file.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--tokens",
|
||||
type=str,
|
||||
|
Loading…
x
Reference in New Issue
Block a user