Update README.md

README.md

@@ -76,6 +76,8 @@ print(prediction[0])
 ```
 
 ## Evaluation
+For the evaluation use the code below:
+```python
 pip install datasets
 pip install transformers
 import torch
@@ -103,6 +105,7 @@ def predict(batch):
         return_tensors="pt", 
         padding=True
     )
+    
     input_values = features.input_values
     attention_mask = features.attention_mask
 
@@ -117,11 +120,9 @@ dataset = dataset.map(speech_file_to_array_fn)
 result = dataset.map(predict, batched=True, batch_size=4)
 wer = load_metric("wer")
 cer = load_metric("cer")
+
 print("WER: {:.2f}".format(100 * wer.compute(predictions=result["prediction"], references=result["reference"])))
 print("CER: {:.2f}".format(100 * cer.compute(predictions=result["prediction"], references=result["reference"])))
-For the evaluation use the code below:
-```python
-?
 ```
 
 *Result (WER)*: