Update svc_inference.py

svc_inference.py

@@ -135,20 +135,28 @@ def svc_infer(model, retrieval: IRetrieval, spk, pit, ppg, vec, hp, device):
 
 
 def main(args):
-    if (args.ppg == None):
-        args.ppg = "svc_tmp.ppg.npy"
+    temp_dir = "/tmp"
+    os.makedirs(temp_dir, exist_ok=True)
+
+    ppg_file = os.path.join(temp_dir, os.path.basename(args.wave) + ".ppg.npy")
+    vec_file = os.path.join(temp_dir, os.path.basename(args.wave) + ".vec.npy")
+    pit_file = os.path.join(temp_dir, os.path.basename(args.wave) + ".pit.csv")
+
+    args.ppg = ppg_file
+    args.vec = vec_file
+    args.pit = pit_file
+
+    if not os.path.exists(ppg_file):
         print(
             f"Auto run : python whisper/inference.py -w {args.wave} -p {args.ppg}")
         os.system(f"python whisper/inference.py -w {args.wave} -p {args.ppg}")
 
-    if (args.vec == None):
-        args.vec = "svc_tmp.vec.npy"
+    if not os.path.exists(vec_file):
         print(
             f"Auto run : python hubert/inference.py -w {args.wave} -v {args.vec}")
         os.system(f"python hubert/inference.py -w {args.wave} -v {args.vec}")
 
-    if (args.pit == None):
-        args.pit = "svc_tmp.pit.csv"
+    if not os.path.exists(pit_file):
         print(
             f"Auto run : python pitch/inference.py -w {args.wave} -p {args.pit}")
         os.system(f"python pitch/inference.py -w {args.wave} -p {args.pit}")