add option to use custom model

2024-01-15 23:28:02 +08:00
parent 076aebf3b6
commit 71d0fe69c6
3 changed files with 61 additions and 14 deletions
@@ -1,5 +1,13 @@
+import argparse
 from whisper_live.server import TranscriptionServer

 if __name__ == "__main__":
    server = TranscriptionServer()
-    server.run("0.0.0.0")
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--model_path', type=str, default=None, help="Custom Faster Whisper Model")
+    args = parser.parse_args()
+    server.run(
+        "0.0.0.0",
+        9090,
+        custom_model_path=args.model_path
+    )