[WIP] - CLI

transformers-cli

@@ -1,14 +1,15 @@
 #!/usr/bin/env python
 from argparse import ArgumentParser
 
+from transformers.commands.serving import ServeCommand
 from transformers.commands.user import UserCommands
 
-
 if __name__ == '__main__':
-    parser = ArgumentParser(description='Transformers CLI tool', usage='transformers-cli <command> [<args>]')
+    parser = ArgumentParser('Transformers CLI tool', usage='transformers-cli <command> [<args>]')
     commands_parser = parser.add_subparsers(help='transformers-cli command helpers')
 
     # Register commands
+    ServeCommand.register_subcommand(commands_parser)
     UserCommands.register_subcommand(commands_parser)
 
     # Let's go