Skip to content

Commit

Permalink
remove whisper-chichewa asr model
Browse files Browse the repository at this point in the history
  • Loading branch information
nikochiko committed Jan 14, 2025
1 parent 4c944fa commit dce634c
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions daras_ai_v2/asr.py
Original file line number Diff line number Diff line change
Expand Up @@ -233,7 +233,6 @@ class AsrModels(Enum):
whisper_hindi_large_v2 = "Whisper Hindi Large v2 (Bhashini)"
whisper_telugu_large_v2 = "Whisper Telugu Large v2 (Bhashini)"
whisper_large_v3 = "Whisper Large v3 (openai)"
whisper_chichewa_large_v3 = "Whisper Large v3 chichewa (opportunity.org)"
gpt_4_o_audio = "GPT-4o Audio (openai)"
nemo_english = "Conformer English (ai4bharat.org)"
nemo_hindi = "Conformer Hindi (ai4bharat.org)"
Expand Down Expand Up @@ -285,7 +284,6 @@ def supports_input_prompt(self) -> bool:
AsrModels.whisper_large_v2: "openai/whisper-large-v2",
AsrModels.whisper_hindi_large_v2: "vasista22/whisper-hindi-large-v2",
AsrModels.whisper_telugu_large_v2: "vasista22/whisper-telugu-large-v2",
AsrModels.whisper_chichewa_large_v3: "dmatekenya/whisper-large-v3-chichewa",
AsrModels.vakyansh_bhojpuri: "Harveenchadha/vakyansh-wav2vec2-bhojpuri-bhom-60",
AsrModels.nemo_english: "https://objectstore.e2enetworks.net/indic-asr-public/checkpoints/conformer/english_large_data_fixed.nemo",
AsrModels.nemo_hindi: "https://objectstore.e2enetworks.net/indic-asr-public/checkpoints/conformer/stt_hi_conformer_ctc_large_v2.nemo",
Expand All @@ -296,7 +294,6 @@ def supports_input_prompt(self) -> bool:
forced_asr_languages = {
AsrModels.whisper_hindi_large_v2: "hi",
AsrModels.whisper_telugu_large_v2: "te",
AsrModels.whisper_chichewa_large_v3: "shona",
AsrModels.vakyansh_bhojpuri: "bho",
AsrModels.nemo_english: "en",
AsrModels.nemo_hindi: "hi",
Expand All @@ -307,7 +304,6 @@ def supports_input_prompt(self) -> bool:
AsrModels.whisper_large_v3: WHISPER_LARGE_V3_SUPPORTED,
AsrModels.whisper_large_v2: WHISPER_LARGE_V2_SUPPORTED,
AsrModels.whisper_telugu_large_v2: {"te"},
AsrModels.whisper_chichewa_large_v3: {"shona"},
AsrModels.whisper_hindi_large_v2: {"hi"},
AsrModels.vakyansh_bhojpuri: {"bho"},
AsrModels.nemo_english: {"en"},
Expand Down

0 comments on commit dce634c

Please sign in to comment.