Update app.py

app.py

@@ -68,8 +68,8 @@ class CNN1DLSTMAudioClassifier(nn.Module):
     
 num_class = 6
 model = CNN1DLSTMAudioClassifier(num_class)
-
-model.load_state_dict(torch.load("best-model-emotion-recognition.bin", weights_only=False))
+device = torch.device('cpu')
+model.load_state_dict(torch.load("best-model-emotion-recognition.bin", weights_only=True, map_location=device))
 model.eval()
 
 def preprocess_single_audio(file_path, sample_rate=16000, n_mels=128, n_fft=2048, hop_length=512):
@@ -135,8 +135,8 @@ def decode_emotion_prediction(prediction_tensor, label_encoder):
     
     return predicted_emotion, confidence
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = model.to(device)
+# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# model = model.to(device)
 
 def predict(wave):
     wave = preprocess_single_audio(wave)