Update README.md

README.md

@@ -40,8 +40,8 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
 #Since marathi is not present on Common Voice, script for reading the below dataset can be picked up from the eval script below
 mr_test_dataset = all_data['test']
 
-processor = Wav2Vec2Processor.from_pretrained("sumedh/wav2vec2-large-xlsr-marathi") 
-model = Wav2Vec2ForCTC.from_pretrained("sumedh/wav2vec2-large-xlsr-marathi") 
+processor = Wav2Vec2Processor.from_pretrained("Tejas2000/SpeechRecog") 
+model = Wav2Vec2ForCTC.from_pretrained("Tejas2000/SpeechRecog") 
 
 resampler = torchaudio.transforms.Resample(48_000, 16_000) #first arg - input sample, second arg - output sample
 # Preprocessing the datasets. We need to read the aduio files as arrays
@@ -77,8 +77,8 @@ all_data = all_data.train_test_split(test_size=0.10,seed=2020) #seed number is i
 mr_test_dataset = all_data['test']
 wer = load_metric("wer")
 
-processor = Wav2Vec2Processor.from_pretrained("sumedh/wav2vec2-large-xlsr-marathi")
-model = Wav2Vec2ForCTC.from_pretrained("sumedh/wav2vec2-large-xlsr-marathi") 
+processor = Wav2Vec2Processor.from_pretrained("Tejas2000/SpeechRecog")
+model = Wav2Vec2ForCTC.from_pretrained("Tejas2000/SpeechRecog") 
 model.to("cuda")
 
 chars_to_ignore_regex = '[\,\?\.\!\-\;\:\"\“]'