From 5cbd5e7f202048928f5a3b7e47ef4d16ca5c40fd Mon Sep 17 00:00:00 2001
From: jhj0517 <97279763+jhj0517@users.noreply.github.com>
Date: Wed, 26 Jun 2024 21:58:39 +0900
Subject: [PATCH] add arg

---
 user-start-webui.bat | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/user-start-webui.bat b/user-start-webui.bat
index d9527590..a1455d05 100644
--- a/user-start-webui.bat
+++ b/user-start-webui.bat
@@ -13,6 +13,7 @@ set WHISPER_TYPE=
 set WHISPER_MODEL_DIR=
 set FASTER_WHISPER_MODEL_DIR=
 set INSANELY_FAST_WHISPER_MODEL_DIR=
+set DIARIZATION_MODEL_DIR=
 
 
 if not "%SERVER_NAME%"=="" (
@@ -51,7 +52,10 @@ if not "%FASTER_WHISPER_MODEL_DIR%"=="" (
 if not "%INSANELY_FAST_WHISPER_MODEL_DIR%"=="" (
     set INSANELY_FAST_WHISPER_MODEL_DIR_ARG=--insanely_fast_whisper_model_dir "%INSANELY_FAST_WHISPER_MODEL_DIR%"
 )
+if not "%DIARIZATION_MODEL_DIR%"=="" (
+    set DIARIZATION_MODEL_DIR_ARG=--diarization_model_dir "%DIARIZATION_MODEL_DIR%"
+)
 
 :: Call the original .bat script with cli arguments
-start-webui.bat %SERVER_NAME_ARG% %SERVER_PORT_ARG% %USERNAME_ARG% %PASSWORD_ARG% %SHARE_ARG% %THEME_ARG% %API_OPEN% %WHISPER_TYPE_ARG% %WHISPER_MODEL_DIR_ARG% %FASTER_WHISPER_MODEL_DIR_ARG% %INSANELY_FAST_WHISPER_MODEL_DIR_ARG%
+start-webui.bat %SERVER_NAME_ARG% %SERVER_PORT_ARG% %USERNAME_ARG% %PASSWORD_ARG% %SHARE_ARG% %THEME_ARG% %API_OPEN% %WHISPER_TYPE_ARG% %WHISPER_MODEL_DIR_ARG% %FASTER_WHISPER_MODEL_DIR_ARG% %INSANELY_FAST_WHISPER_MODEL_DIR_ARG% %DIARIZATION_MODEL_DIR_ARG%
 pause
\ No newline at end of file