File size: 1,398 Bytes
c0551d3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | from argparse import ArgumentParser
from . import BaseAutoTrainCommand
def run_app_command_factory(args):
return RunAutoTrainAppCommand(
args.port,
args.host,
args.task,
)
class RunAutoTrainAppCommand(BaseAutoTrainCommand):
@staticmethod
def register_subcommand(parser: ArgumentParser):
run_app_parser = parser.add_parser(
"app",
description="✨ Run AutoTrain app",
)
run_app_parser.add_argument(
"--port",
type=int,
default=7860,
help="Port to run the app on",
required=False,
)
run_app_parser.add_argument(
"--host",
type=str,
default="127.0.0.1",
help="Host to run the app on",
required=False,
)
run_app_parser.add_argument(
"--task",
type=str,
required=False,
help="Task to run",
)
run_app_parser.set_defaults(func=run_app_command_factory)
def __init__(self, port, host, task):
self.port = port
self.host = host
self.task = task
def run(self):
if self.task == "dreambooth":
from ..dreambooth_app import main
else:
from ..app import main
demo = main()
demo.queue(concurrency_count=10).launch()
|