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)
|