add ptm

run_demo_multi_models.py

@@ -8,6 +8,9 @@ from transformers import pipeline
 
 DEFAULT_MODEL_NAME = "bhuang/whisper-medium-cv11-french-case-punctuation"
 MODEL_NAMES = [
+    "openai/whisper-small",
+    "openai/whisper-medium",
+    "openai/whisper-large-v2",
     "bhuang/whisper-small-cv11-french",
     "bhuang/whisper-small-cv11-french-case-punctuation",
     "bhuang/whisper-medium-cv11-french",
@@ -105,11 +108,7 @@ mf_transcribe = gr.Interface(
     layout="horizontal",
     theme="huggingface",
     title="Whisper Demo: Transcribe Audio",
-    description=(
-        "Transcribe long-form microphone or audio inputs with the click of a button! Demo uses the the fine-tuned"
-        f" checkpoint [{DEFAULT_MODEL_NAME}](https://huggingface.co/{DEFAULT_MODEL_NAME}) and 🤗 Transformers to transcribe audio files"
-        " of arbitrary length."
-    ),
+    description="Transcribe long-form microphone or audio inputs with the click of a button!",
     allow_flagging="never",
 )
 
@@ -123,11 +122,7 @@ yt_transcribe = gr.Interface(
     layout="horizontal",
     theme="huggingface",
     title="Whisper Demo: Transcribe YouTube",
-    description=(
-        "Transcribe long-form YouTube videos with the click of a button! Demo uses the the fine-tuned checkpoint:"
-        f" [{DEFAULT_MODEL_NAME}](https://huggingface.co/{DEFAULT_MODEL_NAME}) and 🤗 Transformers to transcribe audio files of"
-        " arbitrary length."
-    ),
+    description="Transcribe long-form YouTube videos with the click of a button!",
     allow_flagging="never",
 )