Update README.md

README.md

@@ -39,10 +39,8 @@ test_ds = test_ds.map(map_to_array)
 def map_to_pred(batch):
     inputs = processor(batch["speech"], sampling_rate=16000, return_tensors="pt", padding="longest")
     input_values = inputs.input_values.to("cuda")
-    #attention_mask = inputs.attention_mask.to("cuda")
     
     with torch.no_grad():
-        #logits = model(input_values, attention_mask=attention_mask).logits
         logits = model(input_values).logits
 
     predicted_ids = torch.argmax(logits, dim=-1)