Hugging Face's logo

Spaces:
raffaelsiregar
/
speech-emotion-recognition
Sleeping

App Files Community
raffaelsiregar commited on
Commit
3b5b2d9
·
verified ·
1 Parent(s): 51d9b91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ class CNN1DLSTMAudioClassifier(nn.Module):
69
  num_class = 6
70
  model = CNN1DLSTMAudioClassifier(num_class)
71
  device = torch.device('cpu')
72
- model.load_state_dict(torch.load("best-model-emotion-recognition.bin", weights_only=True, map_location=device))
73
  model.eval()
74
 
75
  def preprocess_single_audio(file_path, sample_rate=16000, n_mels=128, n_fft=2048, hop_length=512):
 
69
  num_class = 6
70
  model = CNN1DLSTMAudioClassifier(num_class)
71
  device = torch.device('cpu')
72
+ model.load_state_dict(torch.load("best-model-emotion-recognition.bin", map_location=device))
73
  model.eval()
74
 
75
  def preprocess_single_audio(file_path, sample_rate=16000, n_mels=128, n_fft=2048, hop_length=512):