fix constructor to receive model_dir

app.py

@@ -26,17 +26,25 @@ class App:
         whisper_type = self.args.whisper_type.lower().strip()
 
         if whisper_type in ["faster_whisper", "faster-whisper", "fasterwhisper"]:
-            whisper_inf = FasterWhisperInference()
+            whisper_inf = FasterWhisperInference(
+                model_dir=self.args.faster_whisper_model_dir
+            )
             whisper_inf.model_dir = self.args.faster_whisper_model_dir
         elif whisper_type in ["whisper"]:
-            whisper_inf = WhisperInference()
+            whisper_inf = WhisperInference(
+                model_dir=self.args.whisper_model_dir
+            )
             whisper_inf.model_dir = self.args.whisper_model_dir
         elif whisper_type in ["insanely_fast_whisper", "insanely-fast-whisper", "insanelyfastwhisper",
                               "insanely_faster_whisper", "insanely-faster-whisper", "insanelyfasterwhisper"]:
-            whisper_inf = InsanelyFastWhisperInference()
+            whisper_inf = InsanelyFastWhisperInference(
+                model_dir=self.args.insanely_fast_whisper_model_dir
+            )
             whisper_inf.model_dir = self.args.insanely_fast_whisper_model_dir
         else:
-            whisper_inf = FasterWhisperInference()
+            whisper_inf = FasterWhisperInference(
+                model_dir=self.args.faster_whisper_model_dir
+            )
             whisper_inf.model_dir = self.args.faster_whisper_model_dir
         return whisper_inf
 

modules/faster_whisper_inference.py

@@ -17,9 +17,11 @@ os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
 
 
 class FasterWhisperInference(WhisperBase):
-    def __init__(self):
+    def __init__(self,
+                 model_dir: str
+                 ):
         super().__init__(
-            model_dir=os.path.join("models", "Whisper", "faster-whisper")
+            model_dir=model_dir
         )
         self.model_paths = self.get_model_paths()
         self.available_models = self.model_paths.keys()

modules/insanely_fast_whisper_inference.py

@@ -15,9 +15,11 @@ from modules.whisper_base import WhisperBase
 
 
 class InsanelyFastWhisperInference(WhisperBase):
-    def __init__(self):
+    def __init__(self,
+                 model_dir: str
+                 ):
         super().__init__(
-            model_dir=os.path.join("models", "Whisper", "insanely_fast_whisper")
+            model_dir=model_dir
         )
         openai_models = whisper.available_models()
         distil_models = ["distil-large-v2", "distil-large-v3", "distil-medium.en", "distil-small.en"]

modules/whisper_Inference.py

@@ -11,9 +11,11 @@ from modules.whisper_parameter import *
 
 
 class WhisperInference(WhisperBase):
-    def __init__(self):
+    def __init__(self,
+                 model_dir: str
+                 ):
         super().__init__(
-            model_dir=os.path.join("models", "Whisper")
+            model_dir=model_dir
         )
 
     def transcribe(self,