Harveenchadha commited on
Commit
6819142
1 Parent(s): 3a6bb54

Minor Fixes

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -50,10 +50,10 @@ import torch
50
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
51
  import argparse
52
 
53
- def parse_transcription(checkpoint_path, wav_file):
54
  # load pretrained model
55
- processor = Wav2Vec2Processor.from_pretrained(checkpoint_path)
56
- model = Wav2Vec2ForCTC.from_pretrained(checkpoint_path)
57
 
58
  # load audio
59
  audio_input, sample_rate = sf.read(wav_file)
@@ -125,7 +125,7 @@ print("WER: {:2f}".format(100 * wer.compute(predictions=result["pred_strings"],
125
 
126
  **Test Result**: 33.17 %
127
 
128
- [**Colab Demo**](https://colab.research.google.com/github/harveenchadha/bol/blob/main/demos/hf/hindi/hf_vakyansh_hindi_him_4200_evaluation_common_voice.ipynb)
129
 
130
  ## Credits
131
  Thanks to Ekstep Foundation for making this possible. The vakyansh team will be open sourcing speech models in all the Indic Languages.
 
50
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
51
  import argparse
52
 
53
+ def parse_transcription(wav_file):
54
  # load pretrained model
55
+ processor = Wav2Vec2Processor.from_pretrained("Harveenchadha/vakyansh-wav2vec2-hindi-him-4200")
56
+ model = Wav2Vec2ForCTC.from_pretrained("Harveenchadha/vakyansh-wav2vec2-hindi-him-4200")
57
 
58
  # load audio
59
  audio_input, sample_rate = sf.read(wav_file)
 
125
 
126
  **Test Result**: 33.17 %
127
 
128
+ [**Colab Evaluation**](https://colab.research.google.com/github/harveenchadha/bol/blob/main/demos/hf/hindi/hf_vakyansh_hindi_him_4200_evaluation_common_voice.ipynb)
129
 
130
  ## Credits
131
  Thanks to Ekstep Foundation for making this possible. The vakyansh team will be open sourcing speech models in all the Indic Languages.