Update cli.py

t5s/cli.py

@@ -124,6 +124,7 @@ def start(arguments):
     model_params = {
         "name": arguments["name"],
         "model_type": arguments["model_type"],
+        "model_name": arguments["model_name"],
 def parse_args(args):
     arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
     return arguments