Spaces:
Sleeping
Sleeping
Update predict.py
Browse files- 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=
|
| 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")
|