Update README.md
Browse files
README.md
CHANGED
@@ -34,6 +34,7 @@ pip install optimum[openvino]
|
|
34 |
2. Run model inference:
|
35 |
|
36 |
```
|
|
|
37 |
from transformers import AutoProcessor
|
38 |
from optimum.intel.openvino import OVModelForSpeechSeq2Seq
|
39 |
|
@@ -44,14 +45,14 @@ model = OVModelForSpeechSeq2Seq.from_pretrained(model_id)
|
|
44 |
dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation", trust_remote_code=True)
|
45 |
sample = dataset[0]
|
46 |
|
47 |
-
input_features =
|
48 |
sample["audio"]["array"],
|
49 |
sampling_rate=sample["audio"]["sampling_rate"],
|
50 |
return_tensors="pt",
|
51 |
).input_features
|
52 |
|
53 |
outputs = model.generate(input_features)
|
54 |
-
text =
|
55 |
print(text)
|
56 |
```
|
57 |
|
|
|
34 |
2. Run model inference:
|
35 |
|
36 |
```
|
37 |
+
from datasets import load_dataset
|
38 |
from transformers import AutoProcessor
|
39 |
from optimum.intel.openvino import OVModelForSpeechSeq2Seq
|
40 |
|
|
|
45 |
dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation", trust_remote_code=True)
|
46 |
sample = dataset[0]
|
47 |
|
48 |
+
input_features = tokenizer(
|
49 |
sample["audio"]["array"],
|
50 |
sampling_rate=sample["audio"]["sampling_rate"],
|
51 |
return_tensors="pt",
|
52 |
).input_features
|
53 |
|
54 |
outputs = model.generate(input_features)
|
55 |
+
text = tokenizer.batch_decode(outputs)[0]
|
56 |
print(text)
|
57 |
```
|
58 |
|