junseok520 commited on
Commit
5121459
·
verified ·
1 Parent(s): 08cc398

Update predict.py

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -13,7 +13,7 @@ warnings.filterwarnings("ignore")
13
  def get_arg():
14
  parser = argparse.ArgumentParser()
15
  parser.add_argument("--mode", required=True, choices=["predict_file", "predict_dir"], type=str, help="predict mode")
16
- parser.add_argument("--ckpt_path", required=False, default="voxsim_wavlm_ecapa.model", type=pathlib.Path, help="path to the model checkpoint")
17
  parser.add_argument("--inp_dir", required=False, default=None, type=pathlib.Path, help="input directory when predict_dir mode")
18
  parser.add_argument("--ref_dir", required=False, default=None, type=pathlib.Path, help="reference directory when predict_dir mode")
19
  parser.add_argument("--inp_path", required=False, default=None, type=pathlib.Path, help="input file when predict_file mode")
 
13
  def get_arg():
14
  parser = argparse.ArgumentParser()
15
  parser.add_argument("--mode", required=True, choices=["predict_file", "predict_dir"], type=str, help="predict mode")
16
+ parser.add_argument("--ckpt_path", required=False, default="voxsim_wavlm_ecapa.model", type=str, help="path to the model checkpoint")
17
  parser.add_argument("--inp_dir", required=False, default=None, type=pathlib.Path, help="input directory when predict_dir mode")
18
  parser.add_argument("--ref_dir", required=False, default=None, type=pathlib.Path, help="reference directory when predict_dir mode")
19
  parser.add_argument("--inp_path", required=False, default=None, type=pathlib.Path, help="input file when predict_file mode")