use whisper for swahili

GooeyAI · Feb 21, 2024 · d5af85f · d5af85f
1 parent 67dee52
commit d5af85f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/daras_ai_v2/bots.py b/daras_ai_v2/bots.py
@@ -569,6 +569,9 @@ def _handle_audio_msg(billing_account_user, bot: BotInterface):
             selected_model = AsrModels.whisper_telugu_large_v2.name
         case "bho":
             selected_model = AsrModels.vakyansh_bhojpuri.name
+        case "sw":
+            selected_model = AsrModels.whisper_large_v2.name
+            language = "sw"
         # case "en":
         #     selected_model = AsrModels.usm.name
         #     language = "am-et"