import transformersimport datasetsimport torch processor = transformers.WhisperProcessor.from_pretrained( "Sunbird/asr-whisper-large-v3-salt")model = transformers.WhisperForConditionalGeneration.from_pretrained( "Sunbird/asr-whisper-large-v3-salt") SALT_LANGUAGE_TOKENS_WHISPER = { 'eng': 50259, # English (Ugandan) 'swa': 50318, # Swahili 'ach': 50357, # Acholi 'lgg': 50356, # Lugbara 'lug': 50355, # Luganda 'nyn': 50354, # Runyankole 'teo': 50353, # Ateso 'xog': 50352, # Lusoga 'ttj': 50351, # Rutooro 'kin': 50350, # Kinyarwanda 'myx': 50349, # Lumasaba} # Get some test audiods = datasets.load_dataset('Sunbird/salt', 'multispeaker-lug', split='test')audio = ds[0]['audio']sample_rate = ds[0]['sample_rate'] # Specify a language from one of the above.lang = 'lug' # Apply the modeldevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")input_features = processor( audio, sampling_rate=sample_rate, return_tensors="pt").input_featuresinput_features = input_features.to(device)predicted_ids = model.to(device).generate( input_features, # Optionally set language=None here instead to auto-detect. language=processor.tokenizer.decode(SALT_LANGUAGE_TOKENS_WHISPER[lang]), forced_decoder_ids=None)transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True) print(transcription)# Ekikoola kya kasooli kya kyenvu wabula langi yaakyo etera okuba eya kitaka wansi.