Steveeeeeeen commited on
Commit
14f0d7e
1 Parent(s): 0308d9e

Update demo.md

Browse files
Files changed (1) hide show
  1. demo.md +2 -2
demo.md CHANGED
@@ -50,8 +50,8 @@ def evaluate_model(ds, pipe, wer_metric):
50
  for idx, sample in enumerate(tqdm(ds, desc="Evaluating", total=len(list(ds)))):
51
  audio_sample = sample["audio"]
52
  transcription = pipe(audio_sample["array"])['text']
53
- # Remove , and . from the transcription
54
- transcription = transcription.replace(",", "").replace(".", "")
55
  wer = wer_metric.compute(predictions=[transcription.upper()], references=[sample["text"].upper()])
56
  wer_scores.append(wer)
57
  wer_results.append({
 
50
  for idx, sample in enumerate(tqdm(ds, desc="Evaluating", total=len(list(ds)))):
51
  audio_sample = sample["audio"]
52
  transcription = pipe(audio_sample["array"])['text']
53
+ # Keep only letter and spaces for evaluation
54
+ transcription = "".join([char for char in transcription if char.isalpha() or char.isspace()])
55
  wer = wer_metric.compute(predictions=[transcription.upper()], references=[sample["text"].upper()])
56
  wer_scores.append(wer)
57
  wer_results.append({