Updating the evaluation code. load_metric is deprecated using load from evaluate instead.

README.md

@@ -62,7 +62,7 @@ processor = Wav2Vec2Processor.from_pretrained(MODEL_NAME)
 model = Wav2Vec2ForCTC.from_pretrained(MODEL_NAME)
 
 #Load the dataset
-from datasets import load_dataset, load_metric, Audio
+from datasets import load_dataset, Audio
 ds=load_dataset("carlosdanielhernandezmena/ravnursson_asr",split='test')
 
 #Downsample to 16kHz
@@ -80,7 +80,8 @@ ds = ds.map(prepare_dataset, remove_columns=ds.column_names,num_proc=1)
 
 #Define the evaluation metric
 import numpy as np
-wer_metric = load_metric("wer")
+from evaluate import load
+wer_metric = load("wer")
 def compute_metrics(pred):
     pred_logits = pred.predictions
     pred_ids = np.argmax(pred_logits, axis=-1)