Update README.md
Browse files
README.md
CHANGED
@@ -59,7 +59,7 @@ import numpy as np
|
|
59 |
from pathlib import Path
|
60 |
|
61 |
device = torch.device("cuda")
|
62 |
-
model_name = "
|
63 |
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
|
64 |
model = Wav2Vec2BertForAudioFrameClassification.from_pretrained(model_name).to(device)
|
65 |
|
|
|
59 |
from pathlib import Path
|
60 |
|
61 |
device = torch.device("cuda")
|
62 |
+
model_name = "classla/wav2vecbert2-filledPause"
|
63 |
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
|
64 |
model = Wav2Vec2BertForAudioFrameClassification.from_pretrained(model_name).to(device)
|
65 |
|