up

run_demo.py

@@ -5,7 +5,7 @@ import pytube as pt
 from transformers import pipeline
 from huggingface_hub import model_info
 
-MODEL_NAME = "bhuang/whisper-medium-cv11-french-case-punctuation"
+MODEL_NAME = "bofenghuang/whisper-medium-cv11-french-punct"
 CHUNK_LENGTH_S = 30
 
 device = 0 if torch.cuda.is_available() else "cpu"

run_demo_multi_models.py

@@ -11,15 +11,15 @@ from transformers.utils.logging import disable_progress_bar
 warnings.filterwarnings("ignore")
 disable_progress_bar()
 
-DEFAULT_MODEL_NAME = "bofenghuang/whisper-medium-cv11-french-case-punctuation"
+DEFAULT_MODEL_NAME = "bofenghuang/whisper-medium-cv11-french-punct"
 MODEL_NAMES = [
     "openai/whisper-small",
     "openai/whisper-medium",
     "openai/whisper-large-v2",
     "bofenghuang/whisper-small-cv11-french",
-    "bofenghuang/whisper-small-cv11-french-case-punctuation",
+    "bofenghuang/whisper-small-cv11-french-punct",
     "bofenghuang/whisper-medium-cv11-french",
-    "bofenghuang/whisper-medium-cv11-french-case-punctuation",
+    "bofenghuang/whisper-medium-cv11-french-punct",
 ]
 CHUNK_LENGTH_S = 30
 MAX_NEW_TOKENS = 225