Spaces:
Running
on
T4
Running
on
T4
bofenghuang
commited on
Commit
·
1ffac88
1
Parent(s):
3e0e1ab
Update run_demo_multi_models.py
Browse files- run_demo_multi_models.py +5 -5
run_demo_multi_models.py
CHANGED
@@ -11,15 +11,15 @@ from transformers.utils.logging import disable_progress_bar
|
|
11 |
warnings.filterwarnings("ignore")
|
12 |
disable_progress_bar()
|
13 |
|
14 |
-
DEFAULT_MODEL_NAME = "
|
15 |
MODEL_NAMES = [
|
16 |
"openai/whisper-small",
|
17 |
"openai/whisper-medium",
|
18 |
"openai/whisper-large-v2",
|
19 |
-
"
|
20 |
-
"
|
21 |
-
"
|
22 |
-
"
|
23 |
]
|
24 |
CHUNK_LENGTH_S = 30
|
25 |
MAX_NEW_TOKENS = 225
|
|
|
11 |
warnings.filterwarnings("ignore")
|
12 |
disable_progress_bar()
|
13 |
|
14 |
+
DEFAULT_MODEL_NAME = "bofenghuang/whisper-medium-cv11-french-case-punctuation"
|
15 |
MODEL_NAMES = [
|
16 |
"openai/whisper-small",
|
17 |
"openai/whisper-medium",
|
18 |
"openai/whisper-large-v2",
|
19 |
+
"bofenghuang/whisper-small-cv11-french",
|
20 |
+
"bofenghuang/whisper-small-cv11-french-case-punctuation",
|
21 |
+
"bofenghuang/whisper-medium-cv11-french",
|
22 |
+
"bofenghuang/whisper-medium-cv11-french-case-punctuation",
|
23 |
]
|
24 |
CHUNK_LENGTH_S = 30
|
25 |
MAX_NEW_TOKENS = 225
|