add model path arg and option

Author: dsikka

src/deepsparse/server/cli.py

@@ -79,7 +79,17 @@
     ),
 )
 
-MODEL_OPTION = click.argument("model_path", type=str, default="default")
+MODEL_ARG = click.argument("model", type=str, default=None, required=False)
+MODEL_OPTION = click.option(
+    "--model_path",
+    type=str,
+    default="default",
+    help=(
+        "The path to a model.onnx file, a model folder containing the model.onnx "
+        "and supporting files, or a SparseZoo model stub. "
+        "If not specified, the default model for the task is used."
+    ),
+)
 
 BATCH_OPTION = click.option(
     "--batch_size",
@@ -143,6 +153,7 @@
 @PORT_OPTION
 @LOG_LEVEL_OPTION
 @HOT_RELOAD_OPTION
+@MODEL_ARG
 @MODEL_OPTION
 @BATCH_OPTION
 @CORES_OPTION
@@ -158,6 +169,7 @@ def main(
     log_level: str,
     hot_reload_config: bool,
     model_path: str,
+    model: str,
     batch_size: int,
     num_cores: int,
     num_workers: int,
@@ -207,6 +219,13 @@ def main(
        ...
     ```
     """
+    # the server cli can take a model argument or --model_path option
+    # if the --model_path option is provided, use that
+    # otherwise if the argument is given and --model_path is not used, use the
+    # argument instead
+    if model and model_path == "default":
+        model_path = model
+
     if integration == INTEGRATION_OPENAI:
         if task is None or task != "text_generation":
             task = "text_generation"