5roop commited on
Commit
7758e25
·
verified ·
1 Parent(s): 132d375

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = "5roop/wav2vecbert2-filledPause"
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