Add asr to CLI option

doccano · Oct 12, 2022 · c26eb8e · c26eb8e
1 parent da85945
commit c26eb8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doccano_client/cli/commands.py b/doccano_client/cli/commands.py
@@ -63,7 +63,7 @@ def main():
 
     # Create a parser for web server.
     parser_predict = subparsers.add_parser("predict", help="see `predict -h`")
-    parser_predict.add_argument("--task", type=str, choices=["ner"], required=True, help="task name")
+    parser_predict.add_argument("--task", type=str, choices=["ner", "asr"], required=True, help="task name")
     parser_predict.add_argument("--project", type=int, required=True, help="project id")
     parser_predict.add_argument("--model", type=str, required=True, help="model path")
     parser_predict.add_argument("--mapping", type=str, required=False, help="mapping file for label type")