updated the README file
Browse files
README.md
CHANGED
@@ -44,8 +44,8 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
|
|
44 |
|
45 |
test_dataset = load_dataset("common_voice", "id", split="test[:2%]")
|
46 |
|
47 |
-
processor = Wav2Vec2Processor.from_pretrained("
|
48 |
-
model = Wav2Vec2ForCTC.from_pretrained("
|
49 |
|
50 |
resampler = torchaudio.transforms.Resample(48_000, 16_000)
|
51 |
|
@@ -83,8 +83,8 @@ import re
|
|
83 |
test_dataset = load_dataset("common_voice", "id", split="test")
|
84 |
wer = load_metric("wer")
|
85 |
|
86 |
-
processor = Wav2Vec2Processor.from_pretrained("
|
87 |
-
model = Wav2Vec2ForCTC.from_pretrained("
|
88 |
model.to("cuda")
|
89 |
|
90 |
chars_to_ignore_regex = '[\,\?\.\!\-\;\:\"\“\%\‘\'\”]'
|
|
|
44 |
|
45 |
test_dataset = load_dataset("common_voice", "id", split="test[:2%]")
|
46 |
|
47 |
+
processor = Wav2Vec2Processor.from_pretrained("indonesian-nlp/wav2vec2-large-xlsr-indonesian-baseline")
|
48 |
+
model = Wav2Vec2ForCTC.from_pretrained("indonesian-nlp/wav2vec2-large-xlsr-indonesian-baseline")
|
49 |
|
50 |
resampler = torchaudio.transforms.Resample(48_000, 16_000)
|
51 |
|
|
|
83 |
test_dataset = load_dataset("common_voice", "id", split="test")
|
84 |
wer = load_metric("wer")
|
85 |
|
86 |
+
processor = Wav2Vec2Processor.from_pretrained("indonesian-nlp/wav2vec2-large-xlsr-indonesian-baselin")
|
87 |
+
model = Wav2Vec2ForCTC.from_pretrained("indonesian-nlp/wav2vec2-large-xlsr-indonesian-baseline")
|
88 |
model.to("cuda")
|
89 |
|
90 |
chars_to_ignore_regex = '[\,\?\.\!\-\;\:\"\“\%\‘\'\”]'
|