jhj0517 commited on
Commit
292ccb4
·
1 Parent(s): ec41bf5

fix arg name

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -4,6 +4,7 @@ import argparse
4
 
5
  from modules.whisper_Inference import WhisperInference
6
  from modules.faster_whisper_inference import FasterWhisperInference
 
7
  from modules.nllb_inference import NLLBInference
8
  from ui.htmls import *
9
  from modules.youtube_manager import get_ytmetas
@@ -30,9 +31,9 @@ class App:
30
  elif whisper_type in ["whisper"]:
31
  whisper_inf = WhisperInference()
32
  whisper_inf.model_dir = self.args.whisper_model_dir
33
- elif whisper_type in ["insanely_whisper", "insanely-whisper", "insanelywhisper"]:
34
- whisper_inf = WhisperInference()
35
- whisper_inf.model_dir = self.args.whisper_model_dir
36
  else:
37
  whisper_inf = FasterWhisperInference()
38
  whisper_inf.model_dir = self.args.faster_whisper_model_dir
 
4
 
5
  from modules.whisper_Inference import WhisperInference
6
  from modules.faster_whisper_inference import FasterWhisperInference
7
+ from modules.insanely_fast_whisper_inference import InsanelyFastWhisperInference
8
  from modules.nllb_inference import NLLBInference
9
  from ui.htmls import *
10
  from modules.youtube_manager import get_ytmetas
 
31
  elif whisper_type in ["whisper"]:
32
  whisper_inf = WhisperInference()
33
  whisper_inf.model_dir = self.args.whisper_model_dir
34
+ elif whisper_type in ["insanely_fast_whisper", "insanely-fast-whisper", "insanelyfastwhisper"]:
35
+ whisper_inf = InsanelyFastWhisperInference()
36
+ whisper_inf.model_dir = self.args.insanely_fast_whisper_model_dir
37
  else:
38
  whisper_inf = FasterWhisperInference()
39
  whisper_inf.model_dir = self.args.faster_whisper_model_dir