add arg

jhj0517 · Jun 26, 2024 · 5cbd5e7 · 5cbd5e7
1 parent ab7c9b0
commit 5cbd5e7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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