From 8b766d0ce18251d86a1f35d996698c17f8ad8458 Mon Sep 17 00:00:00 2001 From: heimoshuiyu Date: Wed, 10 Jul 2024 22:14:47 +0800 Subject: [PATCH] Update model and add new arguments --- whisper_fastapi.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/whisper_fastapi.py b/whisper_fastapi.py index 7f22cd7..7dbf421 100644 --- a/whisper_fastapi.py +++ b/whisper_fastapi.py @@ -27,9 +27,11 @@ from prometheus_fastapi_instrumentator import Instrumentator parser = argparse.ArgumentParser() parser.add_argument("--host", default="0.0.0.0", type=str) parser.add_argument("--port", default=5000, type=int) -parser.add_argument("--model", default="large-v2", type=str) +parser.add_argument("--model", default="large-v3", type=str) parser.add_argument("--device", default="auto", type=str) parser.add_argument("--cache_dir", default=None, type=str) +parser.add_argument("--local_files_only", default=False, type=bool) +parser.add_argument("--threads", default=4, type=int) args = parser.parse_args() app = FastAPI() # Instrument your app with default metrics and expose the metrics @@ -41,9 +43,9 @@ transcriber = Transcribe( device=args.device, device_index=0, compute_type="default", - threads=1, + threads=args.threads, cache_directory=args.cache_dir, - local_files_only=False, + local_files_only=args.local_files_only, ) print("Model loaded!")