atsushieee commited on
Commit
8bec389
Ā·
1 Parent(s): e417f48

Update svc_inference.py

Browse files
Files changed (1) hide show
  1. svc_inference.py +4 -3
svc_inference.py CHANGED
@@ -135,9 +135,10 @@ def svc_infer(model, retrieval: IRetrieval, spk, pit, ppg, vec, hp, device):
135
 
136
 
137
  def main(args):
138
- ppg_file = args.wave.replace('.wav', '.ppg.npy')
139
- vec_file = args.wave.replace('.wav', '.vec.npy')
140
- pit_file = args.wave.replace('.wav', '.pit.csv')
 
141
 
142
  args.ppg = ppg_file
143
  args.vec = vec_file
 
135
 
136
 
137
  def main(args):
138
+ base_name = os.path.splitext(args.wave)[0] # ę‹”å¼µå­ć‚’é™¤ć„ćŸćƒ•ć‚”ć‚¤ćƒ«åć‚’å–å¾—
139
+ ppg_file = base_name + '.ppg.npy'
140
+ vec_file = base_name + '.vec.npy'
141
+ pit_file = base_name + '.pit.csv'
142
 
143
  args.ppg = ppg_file
144
  args.vec = vec_file