Update token2wav.py

token2wav.py

@@ -4,6 +4,7 @@ import torch
 import torchaudio
 import s3tokenizer
 import onnxruntime
+from huggingface_hub import hf_hub_download
 
 import torchaudio.compliance.kaldi as kaldi
 from flashcosyvoice.modules.hifigan import HiFTGenerator
@@ -21,7 +22,8 @@ class Token2wav():
         option = onnxruntime.SessionOptions()
         option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
         option.intra_op_num_threads = 1
-        self.spk_model = onnxruntime.InferenceSession(f"{model_path}/campplus.onnx", sess_options=option, providers=["CPUExecutionProvider"])
+        path_onnx = hf_hub_download(repo_id="stepfun-ai/Step-Audio-2-mini", filename="token2wav/campplus.onnx")
+        self.spk_model = onnxruntime.InferenceSession("path_onnx", sess_options=option, providers=["CPUExecutionProvider"])
 
         with open(f"{model_path}/flow.yaml", "r") as f:
             configs = load_hyperpyyaml(f)