mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-06-02 13:29:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add argparse group

Browse source
This commit is contained in:
Yuta Hayashibe 2022-10-02 21:13:36 +09:00
parent f5bd88ab6b
commit 235159b8cf
1 changed files with 47 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

86
whispering/cli.py
View file

@ -84,81 +84,89 @@ def transcribe_from_mic(
def get_opts() -> argparse.Namespace: def get_opts() -> argparse.Namespace:
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument(
group_model = parser.add_argument_group("Whisper model options")
group_model.add_argument(
"--model",
type=str,
choices=available_models(),
)
group_model.add_argument(
"--language", "--language",
type=str, type=str,
default=None, default=None,
choices=sorted(LANGUAGES.keys()) choices=sorted(LANGUAGES.keys())
+ sorted([k.title() for k in TO_LANGUAGE_CODE.keys()]), + sorted([k.title() for k in TO_LANGUAGE_CODE.keys()]),
) )
parser.add_argument( group_model.add_argument(
"--model",
type=str,
choices=available_models(),
)
parser.add_argument(
"--device", "--device",
default="cuda" if torch.cuda.is_available() else "cpu", default="cuda" if torch.cuda.is_available() else "cpu",
help="device to use for PyTorch inference", help="device to use for PyTorch inference",
) )
parser.add_argument(
group_ws = parser.add_argument_group("WebSocket options")
group_ws.add_argument(
"--host",
default="0.0.0.0",
help="host of websocker server",
)
group_ws.add_argument(
"--port",
type=int,
help="Port number of websocker server",
)
group_ctx = parser.add_argument_group("Parsing options")
group_ctx.add_argument(
"--beam_size", "--beam_size",
"-b", "-b",
type=int, type=int,
default=5, default=5,
) )
parser.add_argument( group_ctx.add_argument(
"--num_block", "--num_block",
"-n", "-n",
type=int, type=int,
default=160, default=160,
help="Number of operation unit", help="Number of operation unit",
) )
parser.add_argument( group_ctx.add_argument(
"--temperature", "--temperature",
"-t", "-t",
type=float, type=float,
action="append", action="append",
default=[], default=[],
) )
group_ctx.add_argument(
parser.add_argument(
"--mic",
)
parser.add_argument(
"--debug",
action="store_true",
)
parser.add_argument(
"--host",
default="0.0.0.0",
help="host of websocker server",
)
parser.add_argument(
"--port",
type=int,
help="Port number of websocker server",
)
parser.add_argument(
"--allow-padding", "--allow-padding",
action="store_true", action="store_true",
) )
parser.add_argument( group_ctx.add_argument(
"--mode",
choices=["client"],
)
parser.add_argument(
"--show-devices",
action="store_true",
)
parser.add_argument(
"--no-progress", "--no-progress",
action="store_true", action="store_true",
) )
parser.add_argument( group_ctx.add_argument(
"--no-vad", "--no-vad",
action="store_true", action="store_true",
) )
group_ctx.add_argument(
"--mode",
choices=["client"],
)
group_misc = parser.add_argument_group("Other options")
group_misc.add_argument(
"--mic",
)
group_misc.add_argument(
"--show-devices",
action="store_true",
)
group_misc.add_argument(
"--debug",
action="store_true",
)
opts = parser.parse_args() opts = parser.parse_args()
if opts.beam_size <= 0: if opts.beam_size <= 0: