Spaces:
Running
Running
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
|
| 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)
|