from transformers.pipelines.pt_utils import KeyDataset | |
from datasets import load_dataset | |
pipe = pipeline(model="hf-internal-testing/tiny-random-wav2vec2", device=0) | |
dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation[:10]") | |
for out in pipe(KeyDataset(dataset, "audio")): | |
print(out) | |
Using pipelines for a webserver | |
Creating an inference engine is a complex topic which deserves it's own | |
page. |