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",
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(
"--uri", help="unix:// or tcp://", default="tcp://0.0.0.0:10300"
)
@@ -34,16 +39,17 @@ async def main() -> None:
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(
"--beam-size",
type=int,
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")
args = parser.parse_args()
@@ -61,6 +67,7 @@ async def main() -> None:
AsrProgram(
name="whisper",
description="Whisper",
version=args.version,
attribution=Attribution(
name="OpenAI",
url="https://github.com/openai/whisper",
@@ -69,6 +76,7 @@ async def main() -> None:
models=[
AsrModel(
name=args.model,
version=args.version,
description=args.model,
attribution=Attribution(
name="whisper",
@@ -100,4 +108,4 @@ if __name__ == "__main__":
try:
asyncio.run(main())
except KeyboardInterrupt:
pass
pass

View File

@@ -89,4 +89,4 @@ class FasterWhisperEventHandler(AsyncEventHandler):
return False
return True
return True