Update app.py

app.py

@@ -67,9 +67,10 @@ class CNN1DLSTMAudioClassifier(nn.Module):
         return x
     
 num_class = 6
-model = CNN1DLSTMAudioClassifier(num_class)
 device = torch.device('cpu')
-model.load_state_dict(torch.load("best-model-emotion-recognition.bin", map_location=device))
+state_dict = torch.load('best-model-emotion-recognition.bin', map_location=device)
+model = CNN1DLSTMAudioClassifier(num_class)
+model.load_state_dict(state_dict)
 model.eval()
 
 def preprocess_single_audio(file_path, sample_rate=16000, n_mels=128, n_fft=2048, hop_length=512):