jhj0517 commited on
Commit
fd091ff
·
unverified ·
2 Parent(s): 8295934 7d4cc15

Merge pull request #310 from jhj0517/feature/add-trubo

Browse files
.gitignore CHANGED
@@ -2,6 +2,8 @@
2
  *.png
3
  *.mp4
4
  *.mp3
 
 
5
  venv/
6
  modules/ui/__pycache__/
7
  outputs/
 
2
  *.png
3
  *.mp4
4
  *.mp3
5
+ .idea/
6
+ .pytest_cache/
7
  venv/
8
  modules/ui/__pycache__/
9
  outputs/
modules/whisper/faster_whisper_inference.py CHANGED
@@ -159,7 +159,7 @@ class FasterWhisperInference(WhisperBase):
159
  ----------
160
  Name list of models
161
  """
162
- model_paths = {model:model for model in whisper.available_models()}
163
  faster_whisper_prefix = "models--Systran--faster-whisper-"
164
 
165
  existing_models = os.listdir(self.model_dir)
 
159
  ----------
160
  Name list of models
161
  """
162
+ model_paths = {model:model for model in faster_whisper.available_models()}
163
  faster_whisper_prefix = "models--Systran--faster-whisper-"
164
 
165
  existing_models = os.listdir(self.model_dir)