Update README.md
Browse files
    	
        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"] | 
| 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])
         | 

