fabiogra commited on
Commit
1cea6a5
1 Parent(s): c88f061

fix: inference arg

Browse files
Files changed (1) hide show
  1. scripts/inference.py +1 -1
scripts/inference.py CHANGED
@@ -16,7 +16,7 @@ def main():
16
  p.add_argument("--pretrained_model", "-P", type=str, default="baseline.pth")
17
  p.add_argument("--input", "-i", required=True)
18
  p.add_argument("--output_dir", "-o", type=str, default="")
19
- p.add_argument("--only_no_vocals", "-n", action="store_true")
20
  args = p.parse_args()
21
 
22
  input_file = args.input
 
16
  p.add_argument("--pretrained_model", "-P", type=str, default="baseline.pth")
17
  p.add_argument("--input", "-i", required=True)
18
  p.add_argument("--output_dir", "-o", type=str, default="")
19
+ p.add_argument("--only_no_vocals", "-n", default=True)
20
  args = p.parse_args()
21
 
22
  input_file = args.input