classla
/

Wav2Vec2BertPrimaryStressAudioFrameClassifier

Audio Classification

audio-frame-classification

Model card Files Files and versions Community

5roop commited on May 29

Commit

b43bf19

·

verified ·

1 Parent(s): 7798656

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -66,7 +66,7 @@ if torch.cuda.is_available():
 else:
     device = torch.device("cpu")
-model_name = "5roop/Wav2Vec2BertPrimaryStressAudioFrameClassifier"
 feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 model = Wav2Vec2BertForAudioFrameClassification.from_pretrained(model_name).to(device)
 # Path to the file, containing the word to be annotated:

 else:
     device = torch.device("cpu")
+model_name = "classla/Wav2Vec2BertPrimaryStressAudioFrameClassifier"
 feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 model = Wav2Vec2BertForAudioFrameClassification.from_pretrained(model_name).to(device)
 # Path to the file, containing the word to be annotated: