anton-l HF staff commited on
Commit
8b565fd
1 Parent(s): 13b2d0d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -59,7 +59,7 @@ model = HubertForCTC.from_pretrained("facebook/hubert-xlarge-ls960-ft")
59
 
60
  ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
61
 
62
- input_values = processor(ds[0]["audio"]["array"],, return_tensors="pt").input_values # Batch size 1
63
  logits = model(input_values).logits
64
  predicted_ids = torch.argmax(logits, dim=-1)
65
  transcription = processor.decode(predicted_ids[0])
59
 
60
  ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
61
 
62
+ input_values = processor(ds[0]["audio"]["array"], return_tensors="pt").input_values # Batch size 1
63
  logits = model(input_values).logits
64
  predicted_ids = torch.argmax(logits, dim=-1)
65
  transcription = processor.decode(predicted_ids[0])