Update README.md

README.md

@@ -36,9 +36,13 @@ inputs = feature_extractor(
     audio_arrays, # List of waveforms in numpy array format
     sampling_rate=feature_extractor.sampling_rate, 
     max_length=int(feature_extractor.sampling_rate * max_duration), 
-    padding=True, 
+    padding='max_length', 
     truncation=True, 
     return_tensors='pt'
 )
-logits = mert(**inputs) # The logits shape is (batch_size, num_classes)
+# The shape of `input_values` is (batch_size, sample_rate * max_duration)
+ = inputs['input_values']
+outputs = mert(**inputs)
+# The shape of `logits` is (batch_size, num_classes)
+logits = outputs['logits']
 ```