makaveli
24 lines
959 B
Python
24 lines
959 B
Python
import librosa
|
|
import os
|
|
|
|
import openvino_genai as ov_genai
|
|
import huggingface_hub as hf_hub
|
|
|
|
|
|
class WhisperOpenVINO(object):
|
|
def __init__(self, model_id="OpenVINO/whisper-tiny-fp16-ov", device="CPU", language="en", task="transcribe"):
|
|
model_path = model_id.split('/')[-1]
|
|
cache_dir = os.path.join(os.path.expanduser("~"), ".cache", "openvino_whisper_models")
|
|
os.makedirs(cache_dir, exist_ok=True)
|
|
model_path = os.path.join(cache_dir, model_path)
|
|
if not os.path.exists(model_path):
|
|
hf_hub.snapshot_download(model_id, local_dir=model_path)
|
|
self.model = ov_genai.WhisperPipeline(str(model_path), device=device)
|
|
self.language = language
|
|
self.task = task
|
|
|
|
def transcribe(self, input_audio):
|
|
outputs = self.model.generate(input_audio, return_timestamps=True, language=self.language, task=self.task)
|
|
outputs = [seg for seg in outputs.chunks]
|
|
return outputs
|