added version parameter

This commit is contained in:
albe873
2024-12-13 23:14:57 +01:00
parent 0bf38c1cd4
commit 554acddc0c
2 changed files with 15 additions and 7 deletions

View File

@@ -21,6 +21,11 @@ async def main() -> None:
default="medium", default="medium",
help="Name of whisper model to use (default medium)", help="Name of whisper model to use (default medium)",
) )
parser.add_argument(
"--language",
default="en",
help="Language to use for inference (default: en)",
)
parser.add_argument( parser.add_argument(
"--uri", help="unix:// or tcp://", default="tcp://0.0.0.0:10300" "--uri", help="unix:// or tcp://", default="tcp://0.0.0.0:10300"
) )
@@ -34,16 +39,17 @@ async def main() -> None:
default="cuda", default="cuda",
help="Device to use for inference (default: cuda)", help="Device to use for inference (default: cuda)",
) )
parser.add_argument(
"--language",
default="en",
help="Default language to set for transcription (default: en)",
)
parser.add_argument( parser.add_argument(
"--beam-size", "--beam-size",
type=int, type=int,
default=5, default=5,
) )
parser.add_argument(
"--version",
default="1.0",
help="Version of the whisper model to use (default: 1.0)",
)
parser.add_argument("--debug", action="store_true", help="Log DEBUG messages") parser.add_argument("--debug", action="store_true", help="Log DEBUG messages")
args = parser.parse_args() args = parser.parse_args()
@@ -61,6 +67,7 @@ async def main() -> None:
AsrProgram( AsrProgram(
name="whisper", name="whisper",
description="Whisper", description="Whisper",
version=args.version,
attribution=Attribution( attribution=Attribution(
name="OpenAI", name="OpenAI",
url="https://github.com/openai/whisper", url="https://github.com/openai/whisper",
@@ -69,6 +76,7 @@ async def main() -> None:
models=[ models=[
AsrModel( AsrModel(
name=args.model, name=args.model,
version=args.version,
description=args.model, description=args.model,
attribution=Attribution( attribution=Attribution(
name="whisper", name="whisper",