Update README.md

README.md

@@ -33,7 +33,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torchaudio
-from transformers import Wav2Vec2Config, AutoModelForAudioClassification, Wav2Vec2FeatureExtractor
+from transformers import Wav2Vec2Config, AutoModel, Wav2Vec2FeatureExtractor
 
 import librosa
 import numpy as np
@@ -65,7 +65,7 @@ def predict(path, sampling_rate):
 TRUST = true
 
 config = Wav2Vec2Config.from_pretrained('Aniemore/wav2vec2-xlsr-53-russian-emotion-recognition', trust_remote_code=TRUST)
-model_ = AutoModelForAudioClassification.from_pretrained("Aniemore/wav2vec2-xlsr-53-russian-emotion-recognition", trust_remote_code=TRUST, config=config)
+model_ = AutoModel.from_pretrained("Aniemore/wav2vec2-xlsr-53-russian-emotion-recognition", trust_remote_code=TRUST, config=config)
 feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("Aniemore/wav2vec2-xlsr-53-russian-emotion-recognition")
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")