amokrov commited on
Commit
9acc5f2
·
verified ·
1 Parent(s): ccbb5d8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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 = processor(
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 = processor.batch_decode(outputs)[0]
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