Update README.md

README.md

@@ -21,7 +21,7 @@ model-index:
     metrics:
        - name: Test WER
          type: wer
-         value: 30.54
+         value: 29.42
 ---
 # Wav2Vec2-Large-XLSR-53-Maltese
 Fine-tuned [facebook/wav2vec2-large-xlsr-53](https://huggingface.co/facebook/wav2vec2-large-xlsr-53) in Maltese using the [Common Voice](https://huggingface.co/datasets/common_voice)
@@ -41,7 +41,7 @@ import sys
 
 model_name = "Akashpb13/xlsr_maltese_wav2vec2"
 device = "cuda"
-chars_to_ignore_regex = '[\,\?\.\!\-\;\:\"\“\%\‘\”\�\)\(\*)]'
+chars_to_ignore_regex = '[\\,\\?\\.\\!\\-\\;\\:\\"\\“\\%\\‘\\”\\�\\)\\(\\*)]'
 
 model = Wav2Vec2ForCTC.from_pretrained(model_name).to(device)
 processor = Wav2Vec2Processor.from_pretrained(model_name)
@@ -70,10 +70,10 @@ def map_to_pred(batch):
     batch["target"] = batch["sentence"]
     return batch
 
-result = ds.map(map_to_pred, batched=True, batch_size=16, remove_columns=list(ds.features.keys()))
+result = ds.map(map_to_pred, batched=True, batch_size=1, remove_columns=list(ds.features.keys()))
 
 wer = load_metric("wer")
 print(wer.compute(predictions=result["predicted"], references=result["target"]))
 
 ```
-**Test Result**: 30.54 %
+**Test Result**: 29.42 %