Update README.md
Browse files
README.md
CHANGED
@@ -96,7 +96,7 @@ test_dataset = load_dataset("common_voice", f"{lang_id}", split="test")
|
|
96 |
cer = load_metric("./cer")
|
97 |
|
98 |
processor = Wav2Vec2Processor.from_pretrained(f"{model_id}")
|
99 |
-
model = Wav2Vec2ForCTC.from_pretrained(f"{
|
100 |
model.to("cuda")
|
101 |
|
102 |
resampler = torchaudio.transforms.Resample(48_000, 16_000)
|
|
|
96 |
cer = load_metric("./cer")
|
97 |
|
98 |
processor = Wav2Vec2Processor.from_pretrained(f"{model_id}")
|
99 |
+
model = Wav2Vec2ForCTC.from_pretrained(f"{model_id}")
|
100 |
model.to("cuda")
|
101 |
|
102 |
resampler = torchaudio.transforms.Resample(48_000, 16_000)
|