remove duplicate code

2024-01-10 14:14:48 +00:00
parent 0f9e93d203
commit 244ca9e6ba
3 changed files with 366 additions and 548 deletions
@@ -13,11 +13,6 @@ if __name__ == "__main__":
    if args.backend == "tensorrt":
        if args.whisper_tensorrt_path is None:
            raise ValueError("Please Provide a valid tensorrt model path")
-        
-        from whisper_live.server_tensorrt import TranscriptionServerTRT

-        server = TranscriptionServerTRT()
-        server.run("0.0.0.0", port=6006, whisper_tensorrt_path=args.whisper_tensorrt_path)
-    else:
-        server = TranscriptionServer()
-        server.run("0.0.0.0")
+    server = TranscriptionServer()
+    server.run("0.0.0.0", port=6006, backend=args.backend, whisper_tensorrt_path=args.whisper_tensorrt_path)