Currently only supports the kitten-tts onnx file, need to make it general
Currently only supports the kitten-tts onnx file, need to make it general