Spaces:
Running
Running
jhj0517
commited on
Commit
·
292ccb4
1
Parent(s):
ec41bf5
fix arg name
Browse files
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 ["
|
34 |
-
whisper_inf =
|
35 |
-
whisper_inf.model_dir = self.args.
|
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
|