from transformers import WhisperProcessor, WhisperForConditionalGeneration
import librosa

processor = WhisperProcessor.from_pretrained("ТВОЙ_USERNAME/whisper-tiny-oilgas-russian")
model = WhisperForConditionalGeneration.from_pretrained("ТВОЙ_USERNAME/whisper-tiny-oilgas-russian")

audio, sr = librosa.load("audio.wav", sr=16000)
inputs = processor(audio, sampling_rate=16000, return_tensors="pt")

generated_ids = model.generate(inputs.input_features, language="russian", task="transcribe")
transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]

Авторы

Провоторов Н. Р. (provotorovnikita@gmail.com, @prog_by_onion, https://vk.com/progbyonion)

N07P/whisper-tiny-oilgas-russian

Whisper Tiny Fine-tuned для нефтегазовой отрасли

Описание

Обучение

Результаты

Использование

Авторы

N07P/
whisper-tiny-oilgas-russian