From 71d207a607e0f5ac55f53457de150e6fd642f910 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8D=8E=E6=99=A8?= <hcljsq@gmail.com>
Date: Sat, 30 Dec 2023 23:19:30 +0800
Subject: [PATCH] feat: add large-v3 model

---
 Audio-Transcription-Chrome/popup.html  | 1 +
 Audio-Transcription-Firefox/popup.html | 1 +
 whisper_live/server.py                 | 4 ++--
 whisper_live/transcriber.py            | 4 ++--
 4 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/Audio-Transcription-Chrome/popup.html b/Audio-Transcription-Chrome/popup.html
index 79f18ca..3fae548 100644
--- a/Audio-Transcription-Chrome/popup.html
+++ b/Audio-Transcription-Chrome/popup.html
@@ -140,6 +140,7 @@
       <option value="small" selected>Small</option>
       <option value="medium">Medium</option>
       <option value="large-v2">Large-v2</option>
+      <option value="large-v3">Large-v3</option>
     </select>
   </div>
 </body>
diff --git a/Audio-Transcription-Firefox/popup.html b/Audio-Transcription-Firefox/popup.html
index 2755a76..30184a5 100644
--- a/Audio-Transcription-Firefox/popup.html
+++ b/Audio-Transcription-Firefox/popup.html
@@ -142,6 +142,7 @@
       <option value="small" selected>Small</option>
       <option value="medium">Medium</option>
       <option value="large-v2">Large-v2</option>
+      <option value="large-v3">Large-v3</option>
     </select>
   </div>
 </body>
diff --git a/whisper_live/server.py b/whisper_live/server.py
index 1fbd531..5106be6 100644
--- a/whisper_live/server.py
+++ b/whisper_live/server.py
@@ -211,7 +211,7 @@ class ServeClient:
         self.data = b""
         self.frames = b""
         self.model_sizes = [
-            "tiny", "base", "small", "medium", "large-v2"
+            "tiny", "base", "small", "medium", "large-v2", "large-v3"
         ]
         self.multilingual = multilingual
         self.model_size = self.get_model_size(model_size)
@@ -277,7 +277,7 @@ class ServeClient:
             )
             return None
         
-        if model_size == "large-v2":
+        if model_size in ["large-v2", "large-v3"]:
             self.multilingual = True
             return model_size
 
diff --git a/whisper_live/transcriber.py b/whisper_live/transcriber.py
index a275878..918dc2a 100644
--- a/whisper_live/transcriber.py
+++ b/whisper_live/transcriber.py
@@ -94,7 +94,7 @@ class WhisperModel:
 
         Args:
           model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en,
-            small, small.en, medium, medium.en, large-v1, large-v2, or large), a path to a converted
+            small, small.en, medium, medium.en, large-v1, large-v2, large-v3, or large), a path to a converted
             model directory, or a CTranslate2-converted Whisper model ID from the Hugging Face Hub.
             When a size or a model ID is configured, the converted model is downloaded
             from the Hugging Face Hub.
@@ -914,7 +914,7 @@ class WhisperModel:
                 words, word_tokens, start_times, end_times, word_probabilities
             )
         ]
-    
+
     def destroy(self):
         del self.model