Update cli.py

t5s/cli.py

@@ -30,6 +30,8 @@ parser_start.add_argument("--split", default=0.001, help="Enter the split requir
 parser_start.add_argument(
     "-n", "--name", default="summarsiation", help="Enter the name of the model"
 )
+parser_start.add_argument(
+    "-mt", "--model_type", default="t5", help="Enter the model type"
 parser_dirs = command_subparser.add_parser(
     "dirs",
     help="Create directories that are ignored by git but required for " "the project",