bofenghuang commited on
Commit
1ffac88
·
1 Parent(s): 3e0e1ab

Update run_demo_multi_models.py

Browse files
Files changed (1) hide show
  1. 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 = "bhuang/whisper-medium-cv11-french-case-punctuation"
15
  MODEL_NAMES = [
16
  "openai/whisper-small",
17
  "openai/whisper-medium",
18
  "openai/whisper-large-v2",
19
- "bhuang/whisper-small-cv11-french",
20
- "bhuang/whisper-small-cv11-french-case-punctuation",
21
- "bhuang/whisper-medium-cv11-french",
22
- "bhuang/whisper-medium-cv11-french-case-punctuation",
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