add dir initialization

modules/whisper/faster_whisper_inference.py

@@ -17,15 +17,20 @@ from modules.whisper.whisper_base import WhisperBase
 
 class FasterWhisperInference(WhisperBase):
     def __init__(self,
-                 model_dir: str,
-                 output_dir: str,
-                 args: Namespace
+                 model_dir: Optional[str] = None,
+                 diarization_model_dir: Optional[str] = None,
+                 output_dir: Optional[str] = None,
                  ):
         super().__init__(
             model_dir=model_dir,
-            output_dir=output_dir,
-            args=args
+            diarization_model_dir=diarization_model_dir,
+            output_dir=output_dir
         )
+        if model_dir is None:
+            model_dir = os.path.join("models", "Whisper", "faster-whisper")
+        self.model_dir = model_dir
+        os.makedirs(self.model_dir, exist_ok=True)
+
         self.model_paths = self.get_model_paths()
         self.device = self.get_device()
         self.available_models = self.model_paths.keys()

modules/whisper/insanely_fast_whisper_inference.py

@@ -17,15 +17,20 @@ from modules.whisper.whisper_base import WhisperBase
 
 class InsanelyFastWhisperInference(WhisperBase):
     def __init__(self,
-                 model_dir: str,
-                 output_dir: str,
-                 args: Namespace
+                 model_dir: Optional[str] = None,
+                 diarization_model_dir: Optional[str] = None,
+                 output_dir: Optional[str] = None,
                  ):
         super().__init__(
             model_dir=model_dir,
             output_dir=output_dir,
-            args=args
+            diarization_model_dir=diarization_model_dir
         )
+        if model_dir is None:
+            model_dir = os.path.join("models", "Whisper", "insanely-fast-whisper")
+        self.model_dir = model_dir
+        os.makedirs(self.model_dir, exist_ok=True)
+
         openai_models = whisper.available_models()
         distil_models = ["distil-large-v2", "distil-large-v3", "distil-medium.en", "distil-small.en"]
         self.available_models = openai_models + distil_models

modules/whisper/whisper_Inference.py

@@ -12,14 +12,14 @@ from modules.whisper.whisper_parameter import *
 
 class WhisperInference(WhisperBase):
     def __init__(self,
-                 model_dir: str,
-                 output_dir: str,
-                 args: Namespace
+                 model_dir: Optional[str] = None,
+                 diarization_model_dir: Optional[str] = None,
+                 output_dir: Optional[str] = None,
                  ):
         super().__init__(
             model_dir=model_dir,
             output_dir=output_dir,
-            args=args
+            diarization_model_dir=diarization_model_dir
         )
 
     def transcribe(self,