Update app.py
Browse files
app.py
CHANGED
@@ -8,24 +8,13 @@ from subtitle_manager import Subtitle
|
|
8 |
from pathlib import Path
|
9 |
import psutil
|
10 |
import pynvml
|
|
|
11 |
|
12 |
logging.basicConfig(level=logging.INFO)
|
13 |
last_model = None
|
14 |
model = None
|
15 |
description = "faster-whisper is a reimplementation of OpenAI's Whisper model using CTranslate2, which is a fast inference engine for Transformer models."
|
16 |
article = "Read the [documentation here](https://github.com/SYSTRAN/faster-whisper)."
|
17 |
-
whisper_models = [
|
18 |
-
"tiny", "tiny.en",
|
19 |
-
"base", "base.en", "Systran/faster-whisper-base.en", "Systran/faster-whisper-base",
|
20 |
-
"small", "small.en", "distil-small.en", "Systran/faster-distil-whisper-small.en",
|
21 |
-
"medium", "medium.en", "distil-medium.en", "Systran/faster-distil-whisper-medium.en", "Systran/faster-whisper-medium",
|
22 |
-
"large",
|
23 |
-
"large-v1", "Systran/faster-whisper-large-v1",
|
24 |
-
"large-v2", "distil-large-v2", "Systran/faster-distil-whisper-large-v2",
|
25 |
-
"large-v3", "distil-large-v3", "Systran/faster-distil-whisper-large-v3", "distil-whisper/distil-large-v3-ct2",
|
26 |
-
"ivrit-ai/whisper-large-v3-turbo-ct2", "deepdml/faster-whisper-large-v3-turbo-ct2",
|
27 |
-
"distil-whisper/distil-large-v3.5-ct2",
|
28 |
-
]
|
29 |
compute_types = [
|
30 |
"auto", "default", "int8", "int8_float32",
|
31 |
"int8_float16", "int8_bfloat16", "int16",
|
|
|
8 |
from pathlib import Path
|
9 |
import psutil
|
10 |
import pynvml
|
11 |
+
from whisper_models import whisper_models
|
12 |
|
13 |
logging.basicConfig(level=logging.INFO)
|
14 |
last_model = None
|
15 |
model = None
|
16 |
description = "faster-whisper is a reimplementation of OpenAI's Whisper model using CTranslate2, which is a fast inference engine for Transformer models."
|
17 |
article = "Read the [documentation here](https://github.com/SYSTRAN/faster-whisper)."
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
compute_types = [
|
19 |
"auto", "default", "int8", "int8_float32",
|
20 |
"int8_float16", "int8_bfloat16", "int16",
|